Spaces:
Runtime error
Runtime error
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 |
|