Update tasks/text.py

tasks/text.py

@@ -63,7 +63,9 @@ async def evaluate_text(request: TextEvaluationRequest):
 
     #load 
     # Step 1: Use Sentence-BERT to convert text to embeddings
-    model = SentenceTransformer("dunzhang/stella_en_400M_v5",trust_remote_code=True)
+    model = SentenceTransformer("dunzhang/stella_en_400M_v5",trust_remote_code=True,
+                                device="cpu",
+                                config_kwargs={"use_memory_efficient_attention": False, "unpad_inputs": False})
  
     # Convert each sentence into a vector representation (embedding)
     embeddings = model.encode(test_dataset['quote'])