Update app.py

app.py

@@ -13,7 +13,7 @@ import tempfile
 import numpy as np
 from PIL import Image
 import random
-
+import gc
 from optimization import optimize_pipeline_
 
 
@@ -64,16 +64,19 @@ pipe.load_lora_weights(
 )
 kwargs = {}
 kwargs["load_into_transformer_2"] = True
-# pipe.load_lora_weights(
-#    "vrgamedevgirl84/Wan14BT2VFusioniX", 
-#    weight_name="FusionX_LoRa/Phantom_Wan_14B_FusionX_LoRA.safetensors", 
-#     adapter_name="phantom_2", **kwargs
-# )
-# pipe.set_adapters(["phantom", "phantom_2"], adapter_weights=[1., 1.])
+pipe.load_lora_weights(
+   "vrgamedevgirl84/Wan14BT2VFusioniX", 
+   weight_name="FusionX_LoRa/Phantom_Wan_14B_FusionX_LoRA.safetensors", 
+    adapter_name="phantom_2", **kwargs
+)
+pipe.set_adapters(["phantom", "phantom_2"], adapter_weights=[1., 1.])
 pipe.fuse_lora(adapter_names=["phantom"], lora_scale=3., components=["transformer"])
-# pipe.fuse_lora(adapter_names=["phantom_2"], lora_scale=1., components=["transformer_2"])
+pipe.fuse_lora(adapter_names=["phantom_2"], lora_scale=1., components=["transformer_2"])
 pipe.unload_lora_weights()
 
+torch.cuda.empty_cache()
+gc.collect()
+
 optimize_pipeline_(pipe,
     image=Image.new('RGB', (LANDSCAPE_WIDTH, LANDSCAPE_HEIGHT)),
     prompt='prompt',