DawnC commited on
Commit
3694832
·
verified ·
1 Parent(s): f7601ee

Update clip_zero_shot_classifier.py

Browse files
Files changed (1) hide show
  1. clip_zero_shot_classifier.py +1 -1
clip_zero_shot_classifier.py CHANGED
@@ -17,7 +17,7 @@ class CLIPZeroShotClassifier:
17
  初始化CLIP零樣本分類器
18
 
19
  Args:
20
- model_name: CLIP模型名稱,默認為"ViT-L/14"
21
  device: 運行設備,None則自動選擇
22
  """
23
  # 設置運行設備
 
17
  初始化CLIP零樣本分類器
18
 
19
  Args:
20
+ model_name: CLIP模型名稱,默認為"ViT-B/16"
21
  device: 運行設備,None則自動選擇
22
  """
23
  # 設置運行設備