Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ def generate_image(image, prompt, guidance_scale, num_steps, lora_name):
|
|
24 |
# Load and fuse base LoRA weights
|
25 |
|
26 |
# Load selected LoRA effect if not using the pretrained base model
|
27 |
-
if
|
28 |
pipeline.load_lora_weights("nicolaus-huang/PhotoDoodle", weight_name=f"{lora_name}.safetensors")
|
29 |
|
30 |
height=768
|
|
|
24 |
# Load and fuse base LoRA weights
|
25 |
|
26 |
# Load selected LoRA effect if not using the pretrained base model
|
27 |
+
if lora_name != 'pretrained':
|
28 |
pipeline.load_lora_weights("nicolaus-huang/PhotoDoodle", weight_name=f"{lora_name}.safetensors")
|
29 |
|
30 |
height=768
|