3DTopia

Runtime error

ysharma HF staff commited on Feb 13

Commit

1854a09

•

1 Parent(s): 8e13399

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,4 +1,6 @@
 import os
 import sys
 import cv2
 import time

 import os
+os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:128'
 import sys
 import cv2
 import time