瀏覽代碼

rename method

Ayanami Rei 2 年之前
父節點
當前提交
3c9afd5ca3
共有 2 個文件被更改,包括 3 次插入3 次删除
  1. 2 2
      modules/models.py
  2. 1 1
      modules/quant_loader.py

+ 2 - 2
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:

+ 1 - 1
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':