fffiloni commited on
Commit
38a65dc
1 Parent(s): 276f3db

Update app_gradio.py

Browse files
Files changed (1) hide show
  1. app_gradio.py +1 -0
app_gradio.py CHANGED
@@ -112,6 +112,7 @@ pipe = TextToVideoSDPipelineModded.from_pretrained(
112
  unet=pipe_inversion.unet,
113
  ).to(device)
114
 
 
115
  def process_video(num_frames, num_seeds, generator, exp_dir, load_name, caption, lambda_):
116
  pipe_inversion.to(device)
117
  id_latents = invert(pipe_inversion, inv, load_name).to(device, dtype=dtype)
 
112
  unet=pipe_inversion.unet,
113
  ).to(device)
114
 
115
+ @torch.no_grad()
116
  def process_video(num_frames, num_seeds, generator, exp_dir, load_name, caption, lambda_):
117
  pipe_inversion.to(device)
118
  id_latents = invert(pipe_inversion, inv, load_name).to(device, dtype=dtype)