rydlrKE commited on
Commit
1e9ec46
·
verified ·
1 Parent(s): c6cdc8c

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.tensor(encoded_text).to(self.get_device())
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