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

Lint

Browse source
This commit is contained in:
oobabooga 2023-12-19 21:36:57 -08:00
parent 2289e9031e
commit fadb295d4d
3 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
modules/models.py
View file

@ -414,8 +414,8 @@ def ExLlamav2_HF_loader(model_name):
def HQQ_loader(model_name):
try:
from hqq.core.quantize import HQQBackend, HQQLinear
from hqq.engine.hf import HQQModelForCausalLM
from hqq.core.quantize import HQQLinear, HQQBackend
except ModuleNotFoundError:
logger.error("HQQ is not installed. You can install it with:\n\npip install hqq")
return None