Преглед на файлове

Fix GPTQ load_quant call on Windows

oobabooga преди 2 години
родител
ревизия
372363bc3d
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      modules/quantized_LLaMA.py

+ 1 - 1
modules/quantized_LLaMA.py

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