Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -157,6 +157,7 @@ def infer_with_lora(input_image, prompt, selected_index, lora_state, custom_lora
|
|
| 157 |
lora_to_use = flux_loras[selected_index]
|
| 158 |
print(f"Loaded {len(flux_loras)} LoRAs from JSON")
|
| 159 |
# Load LoRA if needed
|
|
|
|
| 160 |
if lora_to_use and lora_to_use != current_lora:
|
| 161 |
try:
|
| 162 |
if current_lora:
|
|
@@ -173,8 +174,9 @@ def infer_with_lora(input_image, prompt, selected_index, lora_state, custom_lora
|
|
| 173 |
print(f"Error loading LoRA: {e}")
|
| 174 |
# Continue without LoRA
|
| 175 |
elif lora_scale != lora_state:
|
| 176 |
-
pipe.set_adapters(["selected_lora"], adapter_weights=[lora_scale])
|
| 177 |
print(f"using already loaded lora: {lora_to_use}, udpated {lora_scale} based on user preference")
|
|
|
|
|
|
|
| 178 |
|
| 179 |
input_image = input_image.convert("RGB")
|
| 180 |
# Add trigger word to prompt
|
|
|
|
| 157 |
lora_to_use = flux_loras[selected_index]
|
| 158 |
print(f"Loaded {len(flux_loras)} LoRAs from JSON")
|
| 159 |
# Load LoRA if needed
|
| 160 |
+
print(f"LoRA to use: {lora_to_use}, Current LoRA: {current_lora}")
|
| 161 |
if lora_to_use and lora_to_use != current_lora:
|
| 162 |
try:
|
| 163 |
if current_lora:
|
|
|
|
| 174 |
print(f"Error loading LoRA: {e}")
|
| 175 |
# Continue without LoRA
|
| 176 |
elif lora_scale != lora_state:
|
|
|
|
| 177 |
print(f"using already loaded lora: {lora_to_use}, udpated {lora_scale} based on user preference")
|
| 178 |
+
pipe.set_adapters(["selected_lora"], adapter_weights=[lora_scale])
|
| 179 |
+
|
| 180 |
|
| 181 |
input_image = input_image.convert("RGB")
|
| 182 |
# Add trigger word to prompt
|