Spaces:
Runtime error
Runtime error
ZeroGPU-safe startup device + llm2vec tensor warning cleanup
Browse files
kimodo/model/llm2vec/llm2vec_wrapper.py
CHANGED
|
@@ -76,5 +76,5 @@ class LLM2VecEncoder:
|
|
| 76 |
encoded_text = encoded_text[0]
|
| 77 |
lengths = lengths[0]
|
| 78 |
|
| 79 |
-
encoded_text = torch.
|
| 80 |
return encoded_text, lengths
|
|
|
|
| 76 |
encoded_text = encoded_text[0]
|
| 77 |
lengths = lengths[0]
|
| 78 |
|
| 79 |
+
encoded_text = torch.as_tensor(encoded_text, device=self.get_device())
|
| 80 |
return encoded_text, lengths
|