MAS-AI-0000 commited on
Commit
e0fafa4
·
verified ·
1 Parent(s): cebcb2c

Update text_embedder.py

Browse files
Files changed (1) hide show
  1. text_embedder.py +2 -2
text_embedder.py CHANGED
@@ -135,7 +135,7 @@ def get_text_embedding(
135
  if _model is None or _tokenizer is None:
136
  return np.zeros((1, 1), dtype=np.float32)
137
 
138
- encoded = _tokenizer.batch_encode_plus(
139
  [text],
140
  return_tensors="pt",
141
  max_length=max_length,
@@ -179,7 +179,7 @@ def get_text_embeddings_batch(
179
  all_embeddings: list[np.ndarray] = []
180
  for i in range(0, len(texts), batch_size):
181
  batch = [str(t) for t in texts[i : i + batch_size]]
182
- encoded = _tokenizer.batch_encode_plus(
183
  batch,
184
  return_tensors="pt",
185
  max_length=max_length,
 
135
  if _model is None or _tokenizer is None:
136
  return np.zeros((1, 1), dtype=np.float32)
137
 
138
+ encoded = _tokenizer(
139
  [text],
140
  return_tensors="pt",
141
  max_length=max_length,
 
179
  all_embeddings: list[np.ndarray] = []
180
  for i in range(0, len(texts), batch_size):
181
  batch = [str(t) for t in texts[i : i + batch_size]]
182
+ encoded = _tokenizer(
183
  batch,
184
  return_tensors="pt",
185
  max_length=max_length,