diff --git a/aiserver.py b/aiserver.py index 1802a502..553f4d04 100644 --- a/aiserver.py +++ b/aiserver.py @@ -17,7 +17,8 @@ os.environ['TOKENIZERS_PARALLELISM'] = 'false' from eventlet import tpool import logging -logging.basicConfig(format='%(levelname)s - %(module)s:%(lineno)d - %(message)s',level=logging.WARNING) +from loguru import logger +logger.debug("That's it, beautiful and simple logging!") logging.getLogger("urllib3").setLevel(logging.ERROR) from os import path, getcwd @@ -66,6 +67,7 @@ try: except: pass import transformers.generation_utils + global tpu_mtj_backend diff --git a/requirements.txt b/requirements.txt index 5dd27bce..4923a140 100644 --- a/requirements.txt +++ b/requirements.txt @@ -15,3 +15,4 @@ accelerate flask-session marshmallow>=3.13 apispec-webframeworks +loguru \ No newline at end of file