Kaynağa Gözat

make argument non case-sensitive

Ayanami Rei 2 yıl önce
ebeveyn
işleme
e1c952c41c
1 değiştirilmiş dosya ile 1 ekleme ve 1 silme
  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:
     elif shared.args.gptq_bits > 0:
         from modules.quant_loader import load_quantized
         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
     # Custom
     else:
     else: