瀏覽代碼

make argument non case-sensitive

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

+ 1 - 1
modules/models.py

@@ -91,7 +91,7 @@ def load_model(model_name):
     elif shared.args.gptq_bits > 0:
         from modules.quant_loader import load_quantized
 
-        model = load_quantized(model_name, shared.args.gptq_model_type)
+        model = load_quantized(model_name, shared.args.gptq_model_type.lower())
 
     # Custom
     else: