Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
00bfb1a
1
Parent(s):
1cdee36
Update app.py
Browse files
app.py
CHANGED
@@ -261,9 +261,11 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, sdxl_lora
|
|
261 |
state_dict_embedding = load_file(embedding_path)
|
262 |
print(state_dict_embedding)
|
263 |
try:
|
|
|
264 |
pipe.load_textual_inversion(state_dict_embedding["clip_l"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
|
265 |
pipe.load_textual_inversion(state_dict_embedding["clip_g"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)
|
266 |
except:
|
|
|
267 |
pipe.load_textual_inversion(state_dict_embedding["text_encoders_0"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
|
268 |
pipe.load_textual_inversion(state_dict_embedding["text_encoders_1"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)
|
269 |
|
|
|
261 |
state_dict_embedding = load_file(embedding_path)
|
262 |
print(state_dict_embedding)
|
263 |
try:
|
264 |
+
pipe.unload_textual_inversion()
|
265 |
pipe.load_textual_inversion(state_dict_embedding["clip_l"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
|
266 |
pipe.load_textual_inversion(state_dict_embedding["clip_g"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)
|
267 |
except:
|
268 |
+
pipe.unload_textual_inversion()
|
269 |
pipe.load_textual_inversion(state_dict_embedding["text_encoders_0"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder, tokenizer=pipe.tokenizer)
|
270 |
pipe.load_textual_inversion(state_dict_embedding["text_encoders_1"], token=["<s0>", "<s1>"], text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)
|
271 |
|