Update app.py

app.py

@@ -15,9 +15,12 @@ from PIL import Image
 import random
 import gc
 from optimization import optimize_pipeline_
+from huggingface_hub import hf_hub_download
 
 
 MODEL_ID = "Wan-AI/Wan2.2-I2V-A14B-Diffusers"
+LORA_REPO_ID = "yeqiu168182/NSFW-22-H-e8"
+LORA_FILENAME = "NSFW-22-H-e8.safetensors"
 
 LANDSCAPE_WIDTH = 832
 LANDSCAPE_HEIGHT = 480
@@ -45,6 +48,10 @@ pipe = WanImageToVideoPipeline.from_pretrained(MODEL_ID,
     torch_dtype=torch.bfloat16,
 ).to('cuda')
 
+#加Lora
+pipe.load_lora_weights(hf_hub_download(repo_id=LORA_REPO_ID, filename=LORA_FILENAME))
+pipe.fuse_lora()
+
 for i in range(3): 
     gc.collect()
     torch.cuda.synchronize()