Upload folder using huggingface_hub

article_embedding/embed.py

@@ -35,7 +35,7 @@ class StellaEmbedder(SentenceTransformerModel):
             "dunzhang/stella_en_400M_v5",
             trust_remote_code=True,
             config_kwargs={"use_memory_efficient_attention": False, "unpad_inputs": False},
-        ).to("mps")
+        )
 
 
 class NvEmbedder(SentenceTransformerModel):
@@ -43,7 +43,7 @@ class NvEmbedder(SentenceTransformerModel):
         self.model = SentenceTransformer(
             "nvidia/NV-Embed-V2",
             trust_remote_code=True,
-        ).to("mps")
+        )
         self.model.max_seq_length = 32768
         self.model.tokenizer.padding_side = "right"
 
@@ -63,7 +63,7 @@ class JasperEmbedder(SentenceTransformerModel):
                 "is_text_encoder": True,
                 "vector_dim": 1024,
             },
-        ).to("mps")
+        )
         self.model.max_seq_length = 1024
 
     def embed(self, documents: list[str]) -> Any: