rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

init

This commit is contained in:
Divided by Zer0 2022-09-11 19:46:51 +02:00
parent 8fce4c192e
commit 88d8f815f5
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
aiserver.py
View File

@ -17,7 +17,8 @@ os.environ['TOKENIZERS_PARALLELISM'] = 'false'
from eventlet import tpool from eventlet import tpool
import logging 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) logging.getLogger("urllib3").setLevel(logging.ERROR)
from os import path, getcwd from os import path, getcwd
@ -66,6 +67,7 @@ try:
except: except:
pass pass
import transformers.generation_utils import transformers.generation_utils
global tpu_mtj_backend global tpu_mtj_backend

1
requirements.txt
View File

@ -15,3 +15,4 @@ accelerate
flask-session flask-session
marshmallow>=3.13 marshmallow>=3.13
apispec-webframeworks apispec-webframeworks
loguru