mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

rename method

Browse source
This commit is contained in:
Ayanami Rei 2023-03-13 20:14:40 +03:00
parent 1b99ed61bc
commit 3c9afd5ca3
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
modules/models.py
View file

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

2
modules/quant_loader.py
View file

@ -10,7 +10,7 @@ sys.path.insert(0, str(Path("repositories/GPTQ-for-LLaMa")))
# 4-bit LLaMA # 4-bit LLaMA
def load_quant(model_name, model_type): def load_quantized(model_name, model_type):
if model_type == 'llama': if model_type == 'llama':
from llama import load_quant from llama import load_quant
elif model_type == 'opt': elif model_type == 'opt':