Spaces:
Running
on
L40S
Running
on
L40S
Set max threads
Browse files
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():
|