multimodalart HF Staff commited on
Commit
d78ad44
·
verified ·
1 Parent(s): b1f43e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -119,8 +119,7 @@ try:
119
  scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(CHECKPOINT_DIR, subfolder="scheduler", torch_dtype=torch_dtype)
120
 
121
  print("Loading DiT model...")
122
- dit = LongCatVideoTransformer3DModel.from_pretrained(CHECKPOINT_DIR, enable_flashattn3=True, enable_flashattn2=False,
123
- enable_xformers=False, subfolder="dit", cp_split_hw=cp_split_hw, torch_dtype=torch_dtype)
124
 
125
  print("Creating LongCatVideoPipeline...")
126
  pipe = LongCatVideoPipeline(
 
119
  scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(CHECKPOINT_DIR, subfolder="scheduler", torch_dtype=torch_dtype)
120
 
121
  print("Loading DiT model...")
122
+ dit = LongCatVideoTransformer3DModel.from_pretrained(CHECKPOINT_DIR, enable_flashattn3=True, enable_flashattn2=False, subfolder="dit", cp_split_hw=cp_split_hw, torch_dtype=torch_dtype)
 
123
 
124
  print("Creating LongCatVideoPipeline...")
125
  pipe = LongCatVideoPipeline(