ensemble

embedding_model_instance.py

@@ -11,9 +11,9 @@ EMBEDDING_MODEL_LARGE = "BAAI/bge-large-en-v1.5"
 
 #EMBEDDING_MODEL = "all-MiniLM-L6-v2"
 
-print(torch.cuda.get_device_name(0))
-print("CUDA available:", torch.cuda.is_available())
-print("Current device:", torch.cuda.current_device())
+# print(torch.cuda.get_device_name(0))
+# print("CUDA available:", torch.cuda.is_available())
+# print("Current device:", torch.cuda.current_device())
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"Using device: {device}")