Update src/CLIPExtractor.py
Browse files- src/CLIPExtractor.py +5 -5
src/CLIPExtractor.py
CHANGED
@@ -11,12 +11,12 @@ class CLIPExtractor:
|
|
11 |
def __init__(self, model_name="openai/clip-vit-large-patch14", cache_dir=None):
|
12 |
|
13 |
# 设置代理环境变量
|
14 |
-
os.environ['HTTP_PROXY'] = 'http://localhost:8234'
|
15 |
-
os.environ['HTTPS_PROXY'] = 'http://localhost:8234'
|
16 |
|
17 |
-
# 设置环境变量
|
18 |
-
os.environ["HF_ENDPOINT"] = "https://hf-api.gitee.com"
|
19 |
-
os.environ["HF_HOME"] = os.path.expanduser("models/")
|
20 |
|
21 |
if not cache_dir:
|
22 |
# 指定缓存目录
|
|
|
11 |
def __init__(self, model_name="openai/clip-vit-large-patch14", cache_dir=None):
|
12 |
|
13 |
# 设置代理环境变量
|
14 |
+
# os.environ['HTTP_PROXY'] = 'http://localhost:8234'
|
15 |
+
# os.environ['HTTPS_PROXY'] = 'http://localhost:8234'
|
16 |
|
17 |
+
# # 设置环境变量
|
18 |
+
# os.environ["HF_ENDPOINT"] = "https://hf-api.gitee.com"
|
19 |
+
# os.environ["HF_HOME"] = os.path.expanduser("models/")
|
20 |
|
21 |
if not cache_dir:
|
22 |
# 指定缓存目录
|