PhyscalX commited on
Commit
1eb61ef
·
1 Parent(s): 04767ba

Set max threads

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