Set max threads
Browse files
app.py
CHANGED
|
@@ -92,7 +92,6 @@ def generate_video(
|
|
| 92 |
device = getattr(pipe, "_offload_device", pipe.device)
|
| 93 |
generator = torch.Generator(device=device).manual_seed(seed)
|
| 94 |
frames = pipe(generator=generator, **args).frames[0]
|
| 95 |
-
torch.cuda.empty_cache() # Clear cache.
|
| 96 |
return export_to_video(frames, fps=12), seed
|
| 97 |
|
| 98 |
|
|
@@ -203,4 +202,4 @@ if __name__ == "__main__":
|
|
| 203 |
],
|
| 204 |
outputs=[result, seed],
|
| 205 |
)
|
| 206 |
-
app.__exit__(), app.launch(share=False, ssr_mode=False)
|
|
|
|
| 92 |
device = getattr(pipe, "_offload_device", pipe.device)
|
| 93 |
generator = torch.Generator(device=device).manual_seed(seed)
|
| 94 |
frames = pipe(generator=generator, **args).frames[0]
|
|
|
|
| 95 |
return export_to_video(frames, fps=12), seed
|
| 96 |
|
| 97 |
|
|
|
|
| 202 |
],
|
| 203 |
outputs=[result, seed],
|
| 204 |
)
|
| 205 |
+
app.__exit__(), app.launch(share=False, ssr_mode=False, max_threads=1)
|