Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-26 01:30:20 +01:00
Code Issues Packages Projects Releases 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
Show Stats 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
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
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():