Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Enable resume training unet/text encoder #48

Merged
merged 2 commits into from
Dec 16, 2022
Merged

Enable resume training unet/text encoder #48

merged 2 commits into from
Dec 16, 2022

Commits on Dec 15, 2022

  1. Enable resume training unet/text encoder 

    New flags --resume_text_encoder --resume_unet accept the paths to .pt files to resume.
Make sure to change the output directory from the previous training session, or else .pt files will be overwritten since training does not resume from previous global step.
    @hdon96
    hdon96 committed Dec 15, 2022
    Configuration menu
    Copy the full SHA
    cded347 View commit details
    Browse the repository at this point in the history

  2. Load weights from .pt with inject_trainable_lora 

    Adds new loras argument to inject_trainable_lora function which accepts path to a .pt file containing previously trained weights.
    @hdon96
    hdon96 committed Dec 15, 2022
    Configuration menu
    Copy the full SHA
    b88d5bd View commit details
    Browse the repository at this point in the history