Spaces:
Sleeping
Sleeping
Yaron Koresh
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -127,8 +127,7 @@ scheduler = DDIMScheduler.from_pretrained(
|
|
| 127 |
|
| 128 |
video_pipe = CogVideoXImageToVideoPipeline.from_pretrained(
|
| 129 |
"THUDM/CogVideoX-5b-I2V",
|
| 130 |
-
torch_dtype=torch.bfloat16
|
| 131 |
-
safety_checker=None
|
| 132 |
).to(device)
|
| 133 |
video_pipe.scheduler = scheduler2
|
| 134 |
|
|
|
|
| 127 |
|
| 128 |
video_pipe = CogVideoXImageToVideoPipeline.from_pretrained(
|
| 129 |
"THUDM/CogVideoX-5b-I2V",
|
| 130 |
+
torch_dtype=torch.bfloat16
|
|
|
|
| 131 |
).to(device)
|
| 132 |
video_pipe.scheduler = scheduler2
|
| 133 |
|