Lint

oobabooga · Dec 20, 2023 · fadb295 · fadb295
1 parent 2289e90
commit fadb295
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/extensions/coqui_tts/script.py b/extensions/coqui_tts/script.py
@@ -10,7 +10,6 @@
 from TTS.utils.synthesizer import Synthesizer
 
 from modules import chat, shared, ui_chat
-from modules.logging_colors import logger
 from modules.ui import create_refresh_button
 from modules.utils import gradio
 

diff --git a/modules/logging_colors.py b/modules/logging_colors.py
@@ -27,10 +27,10 @@ def emit(self, record):
         def get(self):
             return self.buffer
 
-    from rich.theme import Theme
-    from rich.logging import RichHandler
     from rich.console import Console
+    from rich.logging import RichHandler
     from rich.pretty import install as pretty_install
+    from rich.theme import Theme
     from rich.traceback import install as traceback_install
 
     level = logging.DEBUG

diff --git a/modules/models.py b/modules/models.py
@@ -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