Update app.py

app.py

@@ -32,16 +32,16 @@ MAX_DURATION = round(MAX_FRAMES_MODEL/FIXED_FPS,1)
 
 vae = AutoencoderKLWan.from_pretrained("Wan-AI/Wan2.2-T2V-A14B-Diffusers", subfolder="vae", torch_dtype=torch.float32)
 pipe = WanPipeline.from_pretrained(MODEL_ID,
-    # 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('cbensimon/Wan2.2-I2V-A14B-bf16-Diffusers',
-    #     subfolder='transformer_2',
-    #     torch_dtype=torch.bfloat16,
-    #     device_map='cuda',
-    # ),
+    transformer=WanTransformer3DModel.from_pretrained('linoyts/Wan2.2-T2V-A14B-Diffusers-BF16',
+        subfolder='transformer',
+        torch_dtype=torch.bfloat16,
+        device_map='cuda',
+    ),
+    transformer_2=WanTransformer3DModel.from_pretrained('linoyts/Wan2.2-T2V-A14B-Diffusers-BF16',
+        subfolder='transformer_2',
+        torch_dtype=torch.bfloat16,
+        device_map='cuda',
+    ),
     vae=vae,
     torch_dtype=torch.bfloat16,
 ).to('cuda')