rename method

TheTerrasque · Mar 13, 2023 · 3c9afd5 · 3c9afd5
1 parent 1b99ed6
commit 3c9afd5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/modules/models.py b/modules/models.py
@@ -89,9 +89,9 @@ def load_model(model_name):
 
     # Quantized model
     elif shared.args.gptq_bits > 0:
-        from modules.quant_loader import load_quant
+        from modules.quant_loader import load_quantized
 
-        model = load_quant(model_name, shared.args.gptq_model_type)
+        model = load_quantized(model_name, shared.args.gptq_model_type)
 
     # Custom
     else:

diff --git a/modules/quant_loader.py b/modules/quant_loader.py
@@ -10,7 +10,7 @@
 
 
 # 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':