Spaces:
Running
on
Zero
Running
on
Zero
Update clip_zero_shot_classifier.py
Browse files
clip_zero_shot_classifier.py
CHANGED
@@ -17,7 +17,7 @@ class CLIPZeroShotClassifier:
|
|
17 |
初始化CLIP零樣本分類器
|
18 |
|
19 |
Args:
|
20 |
-
model_name: CLIP模型名稱,默認為"ViT-
|
21 |
device: 運行設備,None則自動選擇
|
22 |
"""
|
23 |
# 設置運行設備
|
|
|
17 |
初始化CLIP零樣本分類器
|
18 |
|
19 |
Args:
|
20 |
+
model_name: CLIP模型名稱,默認為"ViT-B/16"
|
21 |
device: 運行設備,None則自動選擇
|
22 |
"""
|
23 |
# 設置運行設備
|