Spaces:
Running
on
Zero
Running
on
Zero
lora
Browse files
app.py
CHANGED
|
@@ -45,6 +45,17 @@ pipe = WanImageToVideoPipeline.from_pretrained(MODEL_ID,
|
|
| 45 |
torch_dtype=torch.bfloat16,
|
| 46 |
).to('cuda')
|
| 47 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 48 |
|
| 49 |
optimize_pipeline_(pipe,
|
| 50 |
image=Image.new('RGB', (LANDSCAPE_WIDTH, LANDSCAPE_HEIGHT)),
|
|
|
|
| 45 |
torch_dtype=torch.bfloat16,
|
| 46 |
).to('cuda')
|
| 47 |
|
| 48 |
+
# load, fuse, unload before compilation
|
| 49 |
+
pipe.load_lora_weights(
|
| 50 |
+
"vrgamedevgirl84/Wan14BT2VFusioniX",
|
| 51 |
+
weight_name="FusionX_LoRa/Phantom_Wan_14B_FusionX_LoRA.safetensors",
|
| 52 |
+
adapter_name="phantom"
|
| 53 |
+
)
|
| 54 |
+
|
| 55 |
+
pipeline.set_adapters(["phantom"], adapter_weights=[0.95])
|
| 56 |
+
pipeline.fuse_lora(adapter_names=["phantom"], lora_scale=1.0)
|
| 57 |
+
pipeline.unload_lora_weights()
|
| 58 |
+
|
| 59 |
|
| 60 |
optimize_pipeline_(pipe,
|
| 61 |
image=Image.new('RGB', (LANDSCAPE_WIDTH, LANDSCAPE_HEIGHT)),
|