Spaces:

jbilcke-hf
/

face-to-all-api

Paused

multimodalart HF staff commited on Apr 2

Commit

00bfb1a

•

1 Parent(s): 1cdee36

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -261,9 +261,11 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, sdxl_lora
             state_dict_embedding = load_file(embedding_path)
             print(state_dict_embedding)
             try:
                 pipe.load_textual_inversion(state_dict_embedding["clip_l"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
                 pipe.load_textual_inversion(state_dict_embedding["clip_g"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)
             except:
                 pipe.load_textual_inversion(state_dict_embedding["text_encoders_0"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
                 pipe.load_textual_inversion(state_dict_embedding["text_encoders_1"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)

             state_dict_embedding = load_file(embedding_path)
             print(state_dict_embedding)
             try:
+                pipe.unload_textual_inversion()
                 pipe.load_textual_inversion(state_dict_embedding["clip_l"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
                 pipe.load_textual_inversion(state_dict_embedding["clip_g"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)
             except:
+                pipe.unload_textual_inversion()
                 pipe.load_textual_inversion(state_dict_embedding["text_encoders_0"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
                 pipe.load_textual_inversion(state_dict_embedding["text_encoders_1"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)