rename method

2023-03-13 20:14:40 +03:00
parent 1b99ed61bc
commit 3c9afd5ca3
2 changed files with 3 additions and 3 deletions
--- a/modules/quant_loader.py
+++ b/modules/quant_loader.py
@@ -10,7 +10,7 @@ sys.path.insert(0, str(Path("repositories/GPTQ-for-LLaMa")))


 # 4-bit LLaMA
-def load_quant(model_name, model_type):
+def load_quantized(model_name, model_type):
    if model_type == 'llama':
        from llama import load_quant
    elif model_type == 'opt':