Spaces:

jbilcke-hf
/

face-to-all-api

Paused

multimodalart HF staff commited on Apr 3

Commit

eab7aad

•

1 Parent(s): 050c639

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -266,8 +266,8 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_stre
                 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_1"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
-                pipe.load_textual_inversion(state_dict_embedding["text_encoders_0"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)
     image = pipe(
         prompt=prompt,

                 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)
     image = pipe(
         prompt=prompt,