hhhwmws commited on
Commit
0c3718a
1 Parent(s): 9b92a87

Update src/CLIPExtractor.py

Browse files
Files changed (1) hide show
  1. 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
  # 指定缓存目录