rahul7star commited on
Commit
cb796fa
·
verified ·
1 Parent(s): ece32ef

Update app_exp.py

Browse files
Files changed (1) hide show
  1. 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=False,
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",