Skip to content

Commit 3c9afd5

Browse files
committed
rename method
1 parent 1b99ed6 commit 3c9afd5

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

modules/models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -89,9 +89,9 @@ def load_model(model_name):
8989

9090
# Quantized model
9191
elif shared.args.gptq_bits > 0:
92-
from modules.quant_loader import load_quant
92+
from modules.quant_loader import load_quantized
9393

94-
model = load_quant(model_name, shared.args.gptq_model_type)
94+
model = load_quantized(model_name, shared.args.gptq_model_type)
9595

9696
# Custom
9797
else:

modules/quant_loader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111

1212
# 4-bit LLaMA
13-
def load_quant(model_name, model_type):
13+
def load_quantized(model_name, model_type):
1414
if model_type == 'llama':
1515
from llama import load_quant
1616
elif model_type == 'opt':

0 commit comments

Comments
 (0)