Spaces:
Running
on
Zero
Running
on
Zero
Update app_exp.py
Browse files- app_exp.py +1 -1
app_exp.py
CHANGED
|
@@ -68,7 +68,7 @@ vae = AutoencoderKLWan.from_pretrained(CHECKPOINT_DIR, subfolder="vae", torch_dt
|
|
| 68 |
scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(CHECKPOINT_DIR, subfolder="scheduler", torch_dtype=torch_dtype)
|
| 69 |
|
| 70 |
dit = LongCatVideoTransformer3DModel.from_pretrained(CHECKPOINT_DIR,
|
| 71 |
-
enable_flashattn3=
|
| 72 |
enable_flashattn2=False,
|
| 73 |
enable_xformers=True,
|
| 74 |
subfolder="dit",
|
|
|
|
| 68 |
scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(CHECKPOINT_DIR, subfolder="scheduler", torch_dtype=torch_dtype)
|
| 69 |
|
| 70 |
dit = LongCatVideoTransformer3DModel.from_pretrained(CHECKPOINT_DIR,
|
| 71 |
+
enable_flashattn3=True,
|
| 72 |
enable_flashattn2=False,
|
| 73 |
enable_xformers=True,
|
| 74 |
subfolder="dit",
|