Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
9e04e04
1
Parent(s):
32b8238
Load changes
Browse files
app.py
CHANGED
|
@@ -51,9 +51,9 @@ def load_and_fuse_pipeline(model_id, lora_filename, pipeline_class, lora_repo_id
|
|
| 51 |
return None
|
| 52 |
|
| 53 |
try:
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
pipe.load_lora_weights(
|
| 57 |
pipe.set_adapters(["fusionx_lora"], adapter_weights=[0.75])
|
| 58 |
pipe.fuse_lora()
|
| 59 |
print(f"✅ FusionX LoRA loaded and fused for {model_id} with a weight of 0.75.")
|
|
|
|
| 51 |
return None
|
| 52 |
|
| 53 |
try:
|
| 54 |
+
print(f"✅ Loading LoRA for {model_id}...")
|
| 55 |
+
# Use the more direct loading method from the diffusers library
|
| 56 |
+
pipe.load_lora_weights(lora_repo_id, weight_name=lora_filename, adapter_name="fusionx_lora")
|
| 57 |
pipe.set_adapters(["fusionx_lora"], adapter_weights=[0.75])
|
| 58 |
pipe.fuse_lora()
|
| 59 |
print(f"✅ FusionX LoRA loaded and fused for {model_id} with a weight of 0.75.")
|