Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -59,7 +59,7 @@ def infer(prompt, video_in, denoise_strength):
|
|
59 |
pipe_xl.vae.enable_slicing()
|
60 |
pipe_xl.scheduler = DPMSolverMultistepScheduler.from_config(pipe_xl.scheduler.config)
|
61 |
pipe_xl.enable_model_cpu_offload()
|
62 |
-
pipe_xl.to("cuda")
|
63 |
video_frames = pipe_xl(prompt, negative_prompt=negative_prompt, video=video_resized, strength=denoise_strength).frames
|
64 |
del pipe_xl
|
65 |
torch.cuda.empty_cache()
|
|
|
59 |
pipe_xl.vae.enable_slicing()
|
60 |
pipe_xl.scheduler = DPMSolverMultistepScheduler.from_config(pipe_xl.scheduler.config)
|
61 |
pipe_xl.enable_model_cpu_offload()
|
62 |
+
#pipe_xl.to("cuda")
|
63 |
video_frames = pipe_xl(prompt, negative_prompt=negative_prompt, video=video_resized, strength=denoise_strength).frames
|
64 |
del pipe_xl
|
65 |
torch.cuda.empty_cache()
|