fffiloni commited on
Commit
a48b85e
1 Parent(s): 1eb3feb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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()