AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases 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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
extensions/coqui_tts/script.py
View File

@ -10,7 +10,6 @@ from TTS.api import TTS
from TTS.utils.synthesizer import Synthesizer from TTS.utils.synthesizer import Synthesizer
from modules import chat, shared, ui_chat from modules import chat, shared, ui_chat
from modules.logging_colors import logger
from modules.ui import create_refresh_button from modules.ui import create_refresh_button
from modules.utils import gradio from modules.utils import gradio

4
modules/logging_colors.py
View File

@ -27,10 +27,10 @@ def setup_logging():
def get(self): def get(self):
return self.buffer return self.buffer
from rich.theme import Theme
from rich.logging import RichHandler
from rich.console import Console from rich.console import Console
from rich.logging import RichHandler
from rich.pretty import install as pretty_install from rich.pretty import install as pretty_install
from rich.theme import Theme
from rich.traceback import install as traceback_install from rich.traceback import install as traceback_install
level = logging.DEBUG level = logging.DEBUG

2
modules/models.py
View File

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