Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -193,8 +193,6 @@ def preset_changed(preset):
|
|
193 |
gr.update(visible=True),
|
194 |
)
|
195 |
|
196 |
-
generator = torch.Generator(device="cpu").manual_seed(seed)
|
197 |
-
|
198 |
# Load models
|
199 |
vae = load_vae(vae_dir)
|
200 |
unet = load_unet(unet_dir)
|
@@ -258,6 +256,7 @@ def generate_video_from_text(
|
|
258 |
"media_items": None,
|
259 |
}
|
260 |
|
|
|
261 |
|
262 |
def gradio_progress_callback(self, step, timestep, kwargs):
|
263 |
progress((step + 1) / num_inference_steps)
|
|
|
193 |
gr.update(visible=True),
|
194 |
)
|
195 |
|
|
|
|
|
196 |
# Load models
|
197 |
vae = load_vae(vae_dir)
|
198 |
unet = load_unet(unet_dir)
|
|
|
256 |
"media_items": None,
|
257 |
}
|
258 |
|
259 |
+
generator = torch.Generator(device="cpu").manual_seed(seed)
|
260 |
|
261 |
def gradio_progress_callback(self, step, timestep, kwargs):
|
262 |
progress((step + 1) / num_inference_steps)
|