Spaces:
Sleeping
Sleeping
Commit
·
859c25a
1
Parent(s):
6e754cf
testing with 2 loras
Browse files
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
|