Removes code specifying the gpu device

defenseunicorns · Aug 27, 2024 · 03186f1 · 03186f1
1 parent d949f74
commit 03186f1
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/packages/text-embeddings/main.py b/packages/text-embeddings/main.py
@@ -11,12 +11,8 @@
     serve,
 )
 
-GPU_ENABLED = (
-    False if os.environ.get("GPU_ENABLED", "False").lower() != "true" else True
-)
-
 model_dir = os.environ.get("LFAI_MODEL_PATH", ".model")
-model = INSTRUCTOR(model_dir, device="cuda" if GPU_ENABLED else "cpu")
+model = INSTRUCTOR(model_dir)
 
 
 class InstructorEmbedding: