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

Avoid a name conflict

Browse source
This commit is contained in:
oobabooga 2023-05-17 11:23:13 -03:00
parent fb91c07191
commit 21ecc3701e
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
modules/models.py
View file

@ -257,17 +257,17 @@ def GPTQ_loader(model_name):
# No monkey patch # No monkey patch
else: else:
from modules.GPTQ_loader import load_quantized import modules.GPTQ_loader
model = load_quantized(model_name) model = modules.GPTQ_loader.load_quantized(model_name)
return model return model
def AutoGPTQ_loader(model_name): def AutoGPTQ_loader(model_name):
from modules.AutoGPTQ_loader import load_quantized import modules.AutoGPTQ_loader
return load_quantized(model_name) return modules.AutoGPTQ_loader.load_quantized(model_name)
def get_max_memory_dict(): def get_max_memory_dict():