multimodalart HF Staff commited on
Commit
fb705d8
·
verified ·
1 Parent(s): 9787259

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, 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 ---
 
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 ---