Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -367,8 +367,9 @@ def generate(prompt,
|
|
367 |
pipe.load_lora_weights(
|
368 |
lora_repo_id,
|
369 |
weight_name=lora_weights_name,
|
|
|
370 |
)
|
371 |
-
pipe.set_adapters(["active_lora"], adapter_weights=[lora_scale_value])
|
372 |
pipe.fuse_lora(adapter_names=["active_lora"],lora_scale=lora_scale_value)
|
373 |
pipe.unload_lora_weights()
|
374 |
print(f"LoRA loaded into main pipe with scale {lora_scale_value}")
|
|
|
367 |
pipe.load_lora_weights(
|
368 |
lora_repo_id,
|
369 |
weight_name=lora_weights_name,
|
370 |
+
adapter_name="active_lora"
|
371 |
)
|
372 |
+
#pipe.set_adapters(["active_lora"], adapter_weights=[lora_scale_value])
|
373 |
pipe.fuse_lora(adapter_names=["active_lora"],lora_scale=lora_scale_value)
|
374 |
pipe.unload_lora_weights()
|
375 |
print(f"LoRA loaded into main pipe with scale {lora_scale_value}")
|