Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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(
|