Spaces:
Sleeping
Sleeping
Update text_embedder.py
Browse files- 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
|
| 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
|
| 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,
|