hjxdl 0.0.14__py3-none-any.whl → 0.0.15__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.
Files changed (6) hide show
  1. hdl/_version.py +2 -2
  2. hdl/utils/llm/embs.py +4 -2
  3. {hjxdl-0.0.14.dist-info → hjxdl-0.0.15.dist-info}/METADATA +1 -1
  4. {hjxdl-0.0.14.dist-info → hjxdl-0.0.15.dist-info}/RECORD +6 -6
  5. {hjxdl-0.0.14.dist-info → hjxdl-0.0.15.dist-info}/WHEEL +0 -0
  6. {hjxdl-0.0.14.dist-info → hjxdl-0.0.15.dist-info}/top_level.txt +0 -0
hdl/_version.py CHANGED
@@ -12,5 +12,5 @@ __version__: str
12
12
  __version_tuple__: VERSION_TUPLE
13
13
  version_tuple: VERSION_TUPLE
14
14
 
15
- __version__ = version = '0.0.14'
16
- __version_tuple__ = version_tuple = (0, 0, 14)
15
+ __version__ = version = '0.0.15'
16
+ __version_tuple__ = version_tuple = (0, 0, 15)
hdl/utils/llm/embs.py CHANGED
@@ -1,4 +1,4 @@
1
- class FlagEmbedder():
1
+ class HFEmbedder():
2
2
  def __init__(
3
3
  self,
4
4
  emb_name: str = "bge",
@@ -41,6 +41,8 @@ class FlagEmbedder():
41
41
  Returns:
42
42
  numpy.ndarray: Encoded representation of the input sentences.
43
43
  """
44
+ if isinstance(sentences, str):
45
+ sentences = [sentences]
44
46
  output = self.model.encode(
45
47
  sentences,
46
48
  return_dense=True,
@@ -68,4 +70,4 @@ class FlagEmbedder():
68
70
  output_1 = self.encode(sentences_1)
69
71
  output_2 = self.encode(sentences_2)
70
72
  similarity = output_1 @ output_2.T
71
- return similarity.item()
73
+ return similarity
{hjxdl-0.0.14.dist-info → hjxdl-0.0.15.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: hjxdl
3
- Version: 0.0.14
3
+ Version: 0.0.15
4
4
  Summary: A collection of functions for Jupyter notebooks
5
5
  Home-page: https://github.com/huluxiaohuowa/hdl
6
6
  Author: Jianxing Hu
{hjxdl-0.0.14.dist-info → hjxdl-0.0.15.dist-info}/RECORD RENAMED
@@ -1,5 +1,5 @@
1
1
  hdl/__init__.py,sha256=5sZZNySv08wwfzJcSDssGTqUn9wlmDsR6R4XB8J8mFM,70
2
- hdl/_version.py,sha256=Z5KVh4ecKbPlzksuEdQrX-Mm8ihZCLuudKr2xk8fqgc,413
2
+ hdl/_version.py,sha256=pzShTmESM-5FMm5QD9BQxtLcs8dS3tIY17WJZlF9dSQ,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
@@ -85,10 +85,10 @@ hdl/utils/general/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
85
85
  hdl/utils/general/glob.py,sha256=8-RCnt6L297wMIfn34ZAMCsGCZUjHG3MGglGZI1cX0g,491
86
86
  hdl/utils/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
87
87
  hdl/utils/llm/chat.py,sha256=NEbWdrPpQqS5ArU7ghA6xuam8wwvWXk3LL_CiNEEuIg,7055
88
- hdl/utils/llm/embs.py,sha256=rkMDaQIVX7cmTEtOhTBzqZYwouOzO3ox0DpJ493BXrs,2124
88
+ hdl/utils/llm/embs.py,sha256=yCFtc25gUFas6kwgOGBFydeaHNyQMq5y1Chxl8TNEUQ,2190
89
89
  hdl/utils/schedulers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
90
90
  hdl/utils/schedulers/norm_lr.py,sha256=bDwCmdEK-WkgxQMFBiMuchv8Mm7C0-GZJ6usm-PQk14,4461
91
- hjxdl-0.0.14.dist-info/METADATA,sha256=0mn1FFKnA5m5jqrIqlh-XioiqkonZ3DXD6I770Q62Xc,543
92
- hjxdl-0.0.14.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
93
- hjxdl-0.0.14.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
94
- hjxdl-0.0.14.dist-info/RECORD,,
91
+ hjxdl-0.0.15.dist-info/METADATA,sha256=uVY_O0levGtYTHXWtCIstsHzulDmM4JLwQ7Xu6uZ5sg,543
92
+ hjxdl-0.0.15.dist-info/WHEEL,sha256=Z4pYXqR_rTB7OWNDYFOm1qRk0RX6GFP2o8LgvP453Hk,91
93
+ hjxdl-0.0.15.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
94
+ hjxdl-0.0.15.dist-info/RECORD,,
{hjxdl-0.0.14.dist-info → hjxdl-0.0.15.dist-info}/WHEEL RENAMED
File without changes