Spaces:
Paused
Paused
Update app_gradio.py
Browse files- 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)
|