Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -88,6 +88,8 @@ def load_b_lora_to_unet(pipe, content_lora_model_id: str = '', style_lora_model_
|
|
88 |
|
89 |
|
90 |
def load_b_loras(content_b_lora, style_b_lora):
|
|
|
|
|
91 |
|
92 |
if content_b_lora != "" and content_b_lora is not None:
|
93 |
# Get instance_prompt a.k.a trigger word
|
@@ -141,8 +143,6 @@ def main(content_b_lora, style_b_lora, prompt, negative_prompt, seed, randomize_
|
|
141 |
width = width,
|
142 |
height = height,
|
143 |
).images[0]
|
144 |
-
|
145 |
-
pipeline.unload_lora_weights()
|
146 |
|
147 |
return image, seed
|
148 |
|
|
|
88 |
|
89 |
|
90 |
def load_b_loras(content_b_lora, style_b_lora):
|
91 |
+
|
92 |
+
pipeline.unload_lora_weights()
|
93 |
|
94 |
if content_b_lora != "" and content_b_lora is not None:
|
95 |
# Get instance_prompt a.k.a trigger word
|
|
|
143 |
width = width,
|
144 |
height = height,
|
145 |
).images[0]
|
|
|
|
|
146 |
|
147 |
return image, seed
|
148 |
|