소스 검색

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: