mirror of
https://github.com/theroyallab/tabbyAPI.git
synced 2026-04-27 01:38:56 +00:00
Logging: Switch to loguru
Loguru is a flexible logger that allows for easier hooking and imports into Rich with no problems. Also makes progress bars stick to the bottom of the terminal window. Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
@@ -2,10 +2,7 @@ import traceback
|
||||
from exllamav2 import ExLlamaV2, ExLlamaV2Tokenizer
|
||||
from exllamav2.generator import ExLlamaV2Sampler
|
||||
from exllamav2.generator.filters import ExLlamaV2Filter, ExLlamaV2PrefixFilter
|
||||
|
||||
from common.logger import init_logger
|
||||
|
||||
logger = init_logger(__name__)
|
||||
from loguru import logger
|
||||
|
||||
|
||||
class OutlinesTokenizerWrapper:
|
||||
|
||||
@@ -15,6 +15,7 @@ from exllamav2 import (
|
||||
ExLlamaV2Lora,
|
||||
)
|
||||
from exllamav2.generator import ExLlamaV2StreamingGenerator, ExLlamaV2Sampler
|
||||
from loguru import logger
|
||||
from typing import List, Optional, Union
|
||||
|
||||
from backends.exllamav2.grammar import ExLlamaV2Grammar
|
||||
@@ -26,9 +27,6 @@ from common.templating import (
|
||||
get_template_from_file,
|
||||
)
|
||||
from common.utils import coalesce, unwrap
|
||||
from common.logger import init_logger
|
||||
|
||||
logger = init_logger(__name__)
|
||||
|
||||
|
||||
class ExllamaV2Container:
|
||||
|
||||
@@ -1,9 +1,6 @@
|
||||
from packaging import version
|
||||
from importlib.metadata import version as package_version
|
||||
|
||||
from common.logger import init_logger
|
||||
|
||||
logger = init_logger(__name__)
|
||||
from loguru import logger
|
||||
|
||||
|
||||
def check_exllama_version():
|
||||
|
||||
Reference in New Issue
Block a user