PhyscalX commited on
Commit
3bc2ffb
·
1 Parent(s): 1eb61ef

Set max threads

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -29,7 +29,6 @@ from diffnext.utils import export_to_image, export_to_video
29
  os.environ["TOKENIZERS_PARALLELISM"] = "true"
30
  # Switch to the allocator optimized for dynamic shape.
31
  os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
32
- os.environ["PYTORCH_ALLOC_CONF"] = "expandable_segments:True"
33
 
34
 
35
  def parse_args():
 
29
  os.environ["TOKENIZERS_PARALLELISM"] = "true"
30
  # Switch to the allocator optimized for dynamic shape.
31
  os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
 
32
 
33
 
34
  def parse_args():