Spaces:
Paused
Paused
Update web-demos/hugging_face/app.py
Browse files
web-demos/hugging_face/app.py
CHANGED
|
@@ -2,6 +2,7 @@ import sys
|
|
| 2 |
sys.path.append("../../")
|
| 3 |
|
| 4 |
import os
|
|
|
|
| 5 |
import json
|
| 6 |
import time
|
| 7 |
import psutil
|
|
|
|
| 2 |
sys.path.append("../../")
|
| 3 |
|
| 4 |
import os
|
| 5 |
+
os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
|
| 6 |
import json
|
| 7 |
import time
|
| 8 |
import psutil
|