Update app.py

app.py

@@ -47,12 +47,12 @@ MIN_DURATION = round(MIN_FRAMES_MODEL/FIXED_FPS,1)
 MAX_DURATION = round(MAX_FRAMES_MODEL/FIXED_FPS,1)
 
 pipe = WanImageToVideoPipeline.from_pretrained(MODEL_ID,
-    transformer=WanTransformer3DModel.from_pretrained('Wan-AI/Wan2.2-I2V-A14B-Diffusers',
+    transformer=WanTransformer3DModel.from_pretrained('cbensimon/Wan2.2-I2V-A14B-bf16-Diffusers',
         subfolder='transformer',
         torch_dtype=torch.bfloat16,
         device_map='cuda',
     ),
-    transformer_2=WanTransformer3DModel.from_pretrained('Wan-AI/Wan2.2-I2V-A14B-Diffusers',
+    transformer_2=WanTransformer3DModel.from_pretrained('cbensimon/Wan2.2-I2V-A14B-bf16-Diffusers',
         subfolder='transformer_2',
         torch_dtype=torch.bfloat16,
         device_map='cuda',
@@ -90,7 +90,7 @@ pipe.load_lora_weights(
 )
 
 
-pipe.set_adapters(["lightx2v", "lora_h", "lightx2v_2", "lora_l"], adapter_weights=[1., 0.3, 1., 0.3])
+pipe.set_adapters(["lightx2v", "lora_h", "lightx2v_2", "lora_l"], adapter_weights=[1., 0.1, 1., 0.3])
 pipe.fuse_lora(adapter_names=["lightx2v"], lora_scale=3., components=["transformer"])
 pipe.fuse_lora(adapter_names=["lora_h"], lora_scale=1., components=["transformer"])
 pipe.fuse_lora(adapter_names=["lightx2v_2"], lora_scale=1., components=["transformer_2"])