hjxdl 0.2.25__py3-none-any.whl → 0.2.26__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- hdl/_version.py +2 -2
- hdl/utils/llm/vis.py +1 -16
- {hjxdl-0.2.25.dist-info → hjxdl-0.2.26.dist-info}/METADATA +1 -1
- {hjxdl-0.2.25.dist-info → hjxdl-0.2.26.dist-info}/RECORD +6 -6
- {hjxdl-0.2.25.dist-info → hjxdl-0.2.26.dist-info}/WHEEL +0 -0
- {hjxdl-0.2.25.dist-info → hjxdl-0.2.26.dist-info}/top_level.txt +0 -0
hdl/_version.py
CHANGED
hdl/utils/llm/vis.py
CHANGED
@@ -133,8 +133,6 @@ class ImgHandler:
|
|
133
133
|
def __init__(
|
134
134
|
self,
|
135
135
|
model_path,
|
136
|
-
db_host = "127.0.0.1",
|
137
|
-
db_port = 8888,
|
138
136
|
conn=None,
|
139
137
|
model_name: str = None,
|
140
138
|
device: str = "cpu",
|
@@ -160,9 +158,7 @@ class ImgHandler:
|
|
160
158
|
self.model_path = model_path
|
161
159
|
self.model_name = model_name
|
162
160
|
|
163
|
-
self.
|
164
|
-
self.db_port = db_port
|
165
|
-
self._db_conn = None
|
161
|
+
self.db_conn = conn
|
166
162
|
self.num_vec_dim = num_vec_dim
|
167
163
|
self.pic_idx_name = "idx:pic_idx"
|
168
164
|
if load_model:
|
@@ -205,17 +201,6 @@ class ImgHandler:
|
|
205
201
|
HF_HUB_PREFIX + self.model_path
|
206
202
|
)
|
207
203
|
|
208
|
-
@property
|
209
|
-
def db_conn(self):
|
210
|
-
"""Establishes a connection to Redis server if not already connected.
|
211
|
-
|
212
|
-
Returns:
|
213
|
-
Redis connection object: A connection to the Redis server.
|
214
|
-
"""
|
215
|
-
if self._db_conn is None:
|
216
|
-
self._db_conn = conn_redis(self.db_host, self.db_port)
|
217
|
-
return self._db_conn
|
218
|
-
|
219
204
|
def get_img_features(
|
220
205
|
self,
|
221
206
|
images,
|
@@ -1,5 +1,5 @@
|
|
1
1
|
hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
|
2
|
-
hdl/_version.py,sha256
|
2
|
+
hdl/_version.py,sha256=-Bz-F1hr7wbrrnyo6EqAHpJHRnGCyOCHoNlwDcFCT1M,413
|
3
3
|
hdl/args/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
4
4
|
hdl/args/loss_args.py,sha256=s7YzSdd7IjD24rZvvOrxLLFqMZQb9YylxKeyelSdrTk,70
|
5
5
|
hdl/controllers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
@@ -132,13 +132,13 @@ hdl/utils/llm/chatgr.py,sha256=GO2G7g6YybduA5VCUuGjvEsJfC_6L7rycSnPeHMcxyM,2820
|
|
132
132
|
hdl/utils/llm/embs.py,sha256=Tf0FOYrOFZp7qQpEPiSCXzlgyHH0X9HVTUtsup74a9E,7174
|
133
133
|
hdl/utils/llm/extract.py,sha256=2sK_WJzmYIc8iuWaM9DA6Nw3_6q1O4lJ5pKpcZo-bBA,6512
|
134
134
|
hdl/utils/llm/llama_chat.py,sha256=watcHGOaz-bv3x-yDucYlGk5f8FiqfFhwWogrl334fk,4387
|
135
|
-
hdl/utils/llm/vis.py,sha256=
|
135
|
+
hdl/utils/llm/vis.py,sha256=M2Xk-2LnOcRge9Aej7ejGZzzLTV82PfI_wo9VX9h-3E,16381
|
136
136
|
hdl/utils/llm/visrag.py,sha256=_PuKtmQIXD5bnmXwDWhTLdzOhgC42JiqdMNb1uKA7n8,9190
|
137
137
|
hdl/utils/schedulers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
138
138
|
hdl/utils/schedulers/norm_lr.py,sha256=bDwCmdEK-WkgxQMFBiMuchv8Mm7C0-GZJ6usm-PQk14,4461
|
139
139
|
hdl/utils/weather/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
140
140
|
hdl/utils/weather/weather.py,sha256=k11o6wM15kF8b9NMlEfrg68ak-SfSYLN3nOOflFUv-I,4381
|
141
|
-
hjxdl-0.2.
|
142
|
-
hjxdl-0.2.
|
143
|
-
hjxdl-0.2.
|
144
|
-
hjxdl-0.2.
|
141
|
+
hjxdl-0.2.26.dist-info/METADATA,sha256=tHMyD2XRvWwLO7NlyzBujX2Bjdx1lnQRxxVWnPwtR_w,836
|
142
|
+
hjxdl-0.2.26.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
|
143
|
+
hjxdl-0.2.26.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
|
144
|
+
hjxdl-0.2.26.dist-info/RECORD,,
|
File without changes
|
File without changes
|