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:
kingbri
2024-03-07 23:20:17 -05:00
committed by Brian Dashore
parent fe0ff240e7
commit 228c227c1e
14 changed files with 110 additions and 119 deletions

View File

@@ -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:

View File

@@ -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:

View File

@@ -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():