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

2023-03-12 17:28:15 +02:00
parent a27f98dbbc
commit 8403152257
1 changed files with 1 additions and 1 deletions
--- a/modules/quantized_LLaMA.py
+++ b/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: