瀏覽代碼

use updated load_quantized

Ayanami Rei 2 年之前
父節點
當前提交
8778b756e6
共有 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.lower())
+        model = load_quantized(model_name)
 
     # Custom
     else: