hjxdl 0.1.71__py3-none-any.whl → 0.1.72__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 +14 -0
- {hjxdl-0.1.71.dist-info → hjxdl-0.1.72.dist-info}/METADATA +1 -1
- {hjxdl-0.1.71.dist-info → hjxdl-0.1.72.dist-info}/RECORD +6 -6
- {hjxdl-0.1.71.dist-info → hjxdl-0.1.72.dist-info}/WHEEL +0 -0
- {hjxdl-0.1.71.dist-info → hjxdl-0.1.72.dist-info}/top_level.txt +0 -0
hdl/_version.py
CHANGED
hdl/utils/llm/vis.py
CHANGED
@@ -68,6 +68,20 @@ class ImgHandler:
|
|
68
68
|
num_vec_dim: int = None,
|
69
69
|
load_model: bool = True,
|
70
70
|
) -> None:
|
71
|
+
"""Initializes the class with the provided parameters.
|
72
|
+
|
73
|
+
Args:
|
74
|
+
model_path (str): Path to the model file.
|
75
|
+
db_host (str): Hostname of the database.
|
76
|
+
db_port (int): Port number of the database.
|
77
|
+
model_name (str, optional): Name of the model. Defaults to None.
|
78
|
+
device (str, optional): Device to run the model on. Defaults to "cpu".
|
79
|
+
num_vec_dim (int, optional): Number of vector dimensions. Defaults to None.
|
80
|
+
load_model (bool, optional): Whether to load the model. Defaults to True.
|
81
|
+
|
82
|
+
Returns:
|
83
|
+
None
|
84
|
+
"""
|
71
85
|
|
72
86
|
self.device = torch.device(device)
|
73
87
|
self.model_path = model_path
|
@@ -1,5 +1,5 @@
|
|
1
1
|
hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
|
2
|
-
hdl/_version.py,sha256=
|
2
|
+
hdl/_version.py,sha256=q-ohSPg6hCd49ZCMCxd2sZVRKz5Qcktg4XZBuNdmSsI,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
|
@@ -131,12 +131,12 @@ hdl/utils/llm/chat.py,sha256=sk7Lw5Oa30k-l2fnJknkMmTc5zkBeEKsR981aeFhH5s,11907
|
|
131
131
|
hdl/utils/llm/embs.py,sha256=Tf0FOYrOFZp7qQpEPiSCXzlgyHH0X9HVTUtsup74a9E,7174
|
132
132
|
hdl/utils/llm/extract.py,sha256=2sK_WJzmYIc8iuWaM9DA6Nw3_6q1O4lJ5pKpcZo-bBA,6512
|
133
133
|
hdl/utils/llm/llama_chat.py,sha256=watcHGOaz-bv3x-yDucYlGk5f8FiqfFhwWogrl334fk,4387
|
134
|
-
hdl/utils/llm/vis.py,sha256=
|
134
|
+
hdl/utils/llm/vis.py,sha256=h9zzDP9ADczxvejUlmLGWNjcdRiMS6FZ_PE29oD0SYc,13727
|
135
135
|
hdl/utils/schedulers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
136
136
|
hdl/utils/schedulers/norm_lr.py,sha256=bDwCmdEK-WkgxQMFBiMuchv8Mm7C0-GZJ6usm-PQk14,4461
|
137
137
|
hdl/utils/weather/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
138
138
|
hdl/utils/weather/weather.py,sha256=k11o6wM15kF8b9NMlEfrg68ak-SfSYLN3nOOflFUv-I,4381
|
139
|
-
hjxdl-0.1.
|
140
|
-
hjxdl-0.1.
|
141
|
-
hjxdl-0.1.
|
142
|
-
hjxdl-0.1.
|
139
|
+
hjxdl-0.1.72.dist-info/METADATA,sha256=wf1HP93Pswqr4_il3n7dWz38gDdkOPEIluEpQdZSAOY,903
|
140
|
+
hjxdl-0.1.72.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
|
141
|
+
hjxdl-0.1.72.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
|
142
|
+
hjxdl-0.1.72.dist-info/RECORD,,
|
File without changes
|
File without changes
|