From 1a61aee75a4f5d581a47208f733958ae3c2ce9a9 Mon Sep 17 00:00:00 2001
From: ChengZi <chen.zhang@zilliz.com>
Date: Wed, 25 May 2022 14:42:25 +0800
Subject: [PATCH] modify factory

---
 __init__.py  | 4 ++--
 clip4clip.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/__init__.py b/__init__.py
index c68e45b..3b16397 100644
--- a/__init__.py
+++ b/__init__.py
@@ -15,6 +15,6 @@
 from .clip4clip import CLIP4Clip
 
 
-def clip4clip(model_name: str, modality: str, weight_path: str, device: str):
-    return CLIP4Clip(model_name, modality, weight_path, device)
+def clip4clip(model_name: str, modality: str, **kwargs):
+    return CLIP4Clip(model_name, modality, **kwargs)
 
diff --git a/clip4clip.py b/clip4clip.py
index 2a8fb24..3cb1b9b 100644
--- a/clip4clip.py
+++ b/clip4clip.py
@@ -37,7 +37,7 @@ class CLIP4Clip(NNOperator):
         self.modality = modality
         if weight_path is None:
             weight_path = str(Path(__file__).parent / 'pytorch_model.bin.1')
-            print('weight_path is None, use default path: {}'.format(weight_path))
+            # print('weight_path is None, use default path: {}'.format(weight_path))
         if device is None:
             self.device = "cuda" if torch.cuda.is_available() else "cpu"
         else: