Abdullah-Habib commited on
Commit
859c25a
·
1 Parent(s): 6e754cf

testing with 2 loras

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -48,8 +48,8 @@ def run_lora(prompt, negative_prompt, cfg_scale, steps, selected_index, schedule
48
  # trigger_word = selected_lora["trigger_word"]
49
 
50
  # Load LoRA weights
51
- pipe.load_lora_weights("Abdullah-Habib/logolora")
52
- pipe.load_lora_weights("Abdullah-Habib/icon-lora")
53
 
54
  # Set scheduler
55
  scheduler_config = pipe.scheduler.config
@@ -100,7 +100,7 @@ def run_lora(prompt, negative_prompt, cfg_scale, steps, selected_index, schedule
100
  width=width,
101
  height=height,
102
  generator=generator,
103
- cross_attention_kwargs={"scale": lora_scale},
104
  ).images[0]
105
 
106
  # Unload LoRA weights
 
48
  # trigger_word = selected_lora["trigger_word"]
49
 
50
  # Load LoRA weights
51
+ pipe.load_lora_weights("Abdullah-Habib/logolora",scale = 1)
52
+ pipe.load_lora_weights("Abdullah-Habib/icon-lora",scale = 0.5)
53
 
54
  # Set scheduler
55
  scheduler_config = pipe.scheduler.config
 
100
  width=width,
101
  height=height,
102
  generator=generator,
103
+ # cross_attention_kwargs={"scale": lora_scale},
104
  ).images[0]
105
 
106
  # Unload LoRA weights