ameerazam08 commited on
Commit
5334972
·
verified ·
1 Parent(s): 47f6862

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 args.lora_name != 'pretrained':
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