Просмотр исходного кода

Fixing compatibility with GPTQ repo commit 2f667f7da051967566a5fb0546f8614bcd3a1ccd. Expects string and breaks on

HideLord 2 лет назад
Родитель
Сommit
8403152257
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      modules/quantized_LLaMA.py

+ 1 - 1
modules/quantized_LLaMA.py

@@ -41,7 +41,7 @@ def load_quantized_LLaMA(model_name):
         print(f"Could not find {pt_model}, exiting...")
         exit()
 
-    model = load_quant(path_to_model, pt_path, bits)
+    model = load_quant(path_to_model, str(pt_path), bits)
 
     # Multi-GPU setup
     if shared.args.gpu_memory: