Commit 890e68aa authored by Fampai's avatar Fampai
Browse files

Fixed minor bug 

when unloading vae during TI training, generating images after
training will error out
parent 3b0127e6

modules/textual_inversion/textual_inversion.py

+1 −0
Original line number Diff line number Diff line
@@ -409,6 +409,7 @@ Last saved image: {html.escape(last_saved_image)}<br/> 


    filename = os.path.join(shared.cmd_opts.embeddings_dir, f'{embedding_name}.pt')

    save_embedding(embedding, checkpoint, embedding_name, filename, remove_cached_checksum=True)

    shared.sd_model.first_stage_model.to(devices.device)



    return embedding, filename

CRA Git | Maintained and supported by SUSTech CRA and CCSE