Update app.py

app.py

@@ -10,6 +10,10 @@ import numpy as np
 from PIL import Image
 import random
 
+os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:512'
+os.environ['HF_HUB_CACHE'] = '/tmp/hub'  # Use temp directory to avoid filling persistent storage
+
+
 MODEL_ID = "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"
 LORA_REPO_ID = "Kijai/WanVideo_comfy"
 LORA_FILENAME = "Wan21_CausVid_14B_T2V_lora_rank32.safetensors"
@@ -25,6 +29,7 @@ pipe.to("cuda")
 causvid_path = hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME)
 pipe.load_lora_weights(causvid_path, adapter_name="causvid_lora")
 pipe.set_adapters(["causvid_lora"], adapter_weights=[0.95])
+pipe.fuse_lora()
 
 MOD_VALUE = 32
 DEFAULT_H_SLIDER_VALUE = 512