Update app.py

app.py

@@ -127,7 +127,13 @@ try:
     scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(CHECKPOINT_DIR, subfolder="scheduler", torch_dtype=torch_dtype)
     
     print("Loading DiT model...")
-    dit = LongCatVideoTransformer3DModel.from_pretrained(CHECKPOINT_DIR, enable_flashattn3=True, enable_flashattn2=False, subfolder="dit", cp_split_hw=cp_split_hw, torch_dtype=torch_dtype)
+    dit = LongCatVideoTransformer3DModel.from_pretrained(CHECKPOINT_DIR,
+                                                         enable_flashattn3=False,
+                                                         enable_flashattn2=False,
+                                                         enable_xformers=True,
+                                                         subfolder="dit",
+                                                         cp_split_hw=cp_split_hw,
+                                                         torch_dtype=torch_dtype)
 
     print("Creating LongCatVideoPipeline...")
     pipe = LongCatVideoPipeline(