Update app.py
Browse files
app.py
CHANGED
@@ -5,9 +5,10 @@ from diffusers.utils import export_to_video
|
|
5 |
|
6 |
# Initialize the diffusion pipeline
|
7 |
pipe = DiffusionPipeline.from_pretrained(
|
8 |
-
"heboya8/text2video-test-2",
|
9 |
torch_dtype=torch.float16,
|
10 |
-
variant="fp16"
|
|
|
11 |
)
|
12 |
|
13 |
# Optimize for GPU memory
|
|
|
5 |
|
6 |
# Initialize the diffusion pipeline
|
7 |
pipe = DiffusionPipeline.from_pretrained(
|
8 |
+
"heboya8/text2video-test-2",
|
9 |
torch_dtype=torch.float16,
|
10 |
+
# variant="fp16",
|
11 |
+
trust_remote_code=True,
|
12 |
)
|
13 |
|
14 |
# Optimize for GPU memory
|