Commit 311354c0 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

fix the issue with training on SD2.0

parent e672cfb0

modules/sd_models.py

+2 −0
Original line number Diff line number Diff line
@@ -228,6 +228,8 @@ def load_model_weights(model, checkpoint_info, vae_file="auto"): 
    model.sd_model_checkpoint = checkpoint_file

    model.sd_checkpoint_info = checkpoint_info



    model.logvar = model.logvar.to(devices.device)  # fix for training



    sd_vae.delete_base_vae()

    sd_vae.clear_loaded_vae()

    vae_file = sd_vae.resolve_vae(checkpoint_file, vae_file=vae_file)

modules/textual_inversion/textual_inversion.py

+1 −2
Original line number Diff line number Diff line
@@ -310,7 +310,6 @@ def train_embedding(embedding_name, learn_rate, batch_size, gradient_step, data_ 
    loss_step = 0

    _loss_step = 0 #internal





    last_saved_file = "<none>"

    last_saved_image = "<none>"

    forced_filename = "<none>"

CRA Git | Maintained and supported by SUSTech CRA and CCSE