Replaces gpu with cuda

defenseunicorns · Aug 27, 2024 · d949f74 · d949f74
1 parent c5c580f
commit d949f74
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/text-embeddings/main.py b/packages/text-embeddings/main.py
@@ -16,7 +16,7 @@
 )
 
 model_dir = os.environ.get("LFAI_MODEL_PATH", ".model")
-model = INSTRUCTOR(model_dir, device="gpu" if GPU_ENABLED else "cpu")
+model = INSTRUCTOR(model_dir, device="cuda" if GPU_ENABLED else "cpu")
 
 
 class InstructorEmbedding: