ghostai1
/

GHOSTSONAFB

Model card Files Files and versions

ghostai1 commited on 8 days ago

Commit

8dadb56

·

verified ·

1 Parent(s): fba37c0

Update appcud12.py

split seg typo changed 32

Files changed (1) hide show

appcud12.py +1 -1

appcud12.py CHANGED Viewed

@@ -17,7 +17,7 @@ import traceback
 warnings.filterwarnings("ignore")
 # Set PYTORCH_CUDA_ALLOC_CONF for CUDA 12
-os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:16"
 # Optimize for CUDA 12
 torch.backends.cudnn.benchmark = False

 warnings.filterwarnings("ignore")
 # Set PYTORCH_CUDA_ALLOC_CONF for CUDA 12
+os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:32"
 # Optimize for CUDA 12
 torch.backends.cudnn.benchmark = False