Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
ef9c19e
1
Parent(s):
1bfb10e
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,14 +9,11 @@ pipe = CogVideoXImageToVideoPipeline.from_pretrained(
|
|
| 9 |
torch_dtype=torch.bfloat16
|
| 10 |
)
|
| 11 |
|
| 12 |
-
pipe.enable_sequential_cpu_offload()
|
| 13 |
pipe.vae.enable_tiling()
|
| 14 |
pipe.vae.enable_slicing()
|
| 15 |
|
| 16 |
@spaces.GPU(duration=250)
|
| 17 |
def generate_video(prompt, image):
|
| 18 |
-
pipe.to("cuda")
|
| 19 |
-
|
| 20 |
video = pipe(
|
| 21 |
prompt=prompt,
|
| 22 |
image=image,
|
|
|
|
| 9 |
torch_dtype=torch.bfloat16
|
| 10 |
)
|
| 11 |
|
|
|
|
| 12 |
pipe.vae.enable_tiling()
|
| 13 |
pipe.vae.enable_slicing()
|
| 14 |
|
| 15 |
@spaces.GPU(duration=250)
|
| 16 |
def generate_video(prompt, image):
|
|
|
|
|
|
|
| 17 |
video = pipe(
|
| 18 |
prompt=prompt,
|
| 19 |
image=image,
|