Update src/alignGenericGGUF.py

src/alignGenericGGUF.py

@@ -34,14 +34,7 @@ from llama_cpp import Llama
 #%%
 start_time = datetime.now()
 
-if os.name == 'nt':  # assume Windows
-    d = 1
-elif os.name == 'posix':
-    if torch.cuda.is_available():  # assume Linux
-        d = 0
-    elif torch.mps.is_available():  # assume macOS
-        d = 1
-dev = ['cuda', 'mps', 'cpu'][d]
+dev = ['cuda', 'mps', 'cpu'][2]  # cpu only
 if dev in ['cuda', 'mps']:
     n_gpu_layers = -1
 else:
@@ -49,7 +42,7 @@ else:
 
 m = 2
 model_name = ['Alibaba-NLP/gte-multilingual-base', 'ibm-granite/granite-embedding-278m-multilingual', 'LaBSEq80', 'LaBSEfp16', 'google/embeddinggemma-300m', 'paraphrase-multilingual-MiniLM-L12-v2'][m]
-model_path = ['', '', '/mnt/d/NLP/koboldcpp/ChristianAzinn/labse-gguf/labse.Q8_0.gguf', '/mnt/d/NLP/koboldcpp/ChristianAzinn/labse-gguf/labse_fp16.gguf', '', ''][m]
+model_path = ['', '', 'src/labse.Q8_0.gguf', '', '', ''][m]
 model_name_short = ['alibaba-gte-multilingual', 'ibm-granite', 'LaBSE-gguf-q80', 'LaBSE-gguf-fp16', 'embeddinggemma-300m','paraphrase'][m]
 #%%
 print(f"Now running bitext mining with transformer model [{model_path}] on device [{dev}]...", flush=True)