diff --git a/modules/loader.py b/modules/loader.py index bb6b120a1..31cadcd56 100644 --- a/modules/loader.py +++ b/modules/loader.py @@ -7,6 +7,7 @@ import urllib3 from modules import timer, errors initialized = False +errors.install() logging.getLogger("DeepSpeed").disabled = True # os.environ.setdefault('OMP_NUM_THREADS', 1) # os.environ.setdefault('MKL_NUM_THREADS', 1) diff --git a/requirements.txt b/requirements.txt index 1aee9cccd..be2442bb9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -63,7 +63,7 @@ pandas==1.5.3 protobuf==3.20.3 pytorch_lightning==1.9.4 tokenizers==0.15.0 -transformers==4.36.0 +transformers==4.36.2 tomesd==0.1.3 urllib3==1.26.15 Pillow==10.1.0