Spaces:
Runtime error
Runtime error
Update all_in_one.py
Browse files- all_in_one.py +1 -0
all_in_one.py
CHANGED
|
@@ -18,6 +18,7 @@ import os
|
|
| 18 |
|
| 19 |
# 设置环境变量
|
| 20 |
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
|
|
|
|
| 21 |
import pickle
|
| 22 |
import cv2
|
| 23 |
import os
|
|
|
|
| 18 |
|
| 19 |
# 设置环境变量
|
| 20 |
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
|
| 21 |
+
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
|
| 22 |
import pickle
|
| 23 |
import cv2
|
| 24 |
import os
|