Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -77,7 +77,7 @@ try:
|
|
| 77 |
scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(CHECKPOINT_DIR, subfolder="scheduler", torch_dtype=torch_dtype)
|
| 78 |
|
| 79 |
print("Loading DiT model...")
|
| 80 |
-
dit = LongCatVideoTransformer3DModel.from_pretrained(CHECKPOINT_DIR, enable_flashattn3=True,
|
| 81 |
enable_xformers=False, subfolder="dit", cp_split_hw=cp_split_hw, torch_dtype=torch_dtype)
|
| 82 |
|
| 83 |
# --- MONKEY PATCH TO FORCE FLASH ATTENTION 3 ---
|
|
|
|
| 77 |
scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(CHECKPOINT_DIR, subfolder="scheduler", torch_dtype=torch_dtype)
|
| 78 |
|
| 79 |
print("Loading DiT model...")
|
| 80 |
+
dit = LongCatVideoTransformer3DModel.from_pretrained(CHECKPOINT_DIR, enable_flashattn3=True, enable_flashattn2=False,
|
| 81 |
enable_xformers=False, subfolder="dit", cp_split_hw=cp_split_hw, torch_dtype=torch_dtype)
|
| 82 |
|
| 83 |
# --- MONKEY PATCH TO FORCE FLASH ATTENTION 3 ---
|