github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Remove transformer log supression from captioner

Browse Source
This commit is contained in:
Jaret Burkett
2026-06-18 10:00:30 -06:00
parent e8c828089a
commit 515b0ea5cd
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions_built_in/captioner/AceStepCaptioner.py
View File

@@ -16,7 +16,7 @@ import transformers
import logging import logging
import warnings import warnings
transformers.logging.set_verbosity_error() # transformers.logging.set_verbosity_error()
warnings.filterwarnings("ignore") warnings.filterwarnings("ignore")
logging.disable(logging.WARNING) logging.disable(logging.WARNING)

2
extensions_built_in/captioner/Ideogram4Captioner.py
View File

@@ -13,7 +13,7 @@ import transformers
import logging import logging
import warnings import warnings
transformers.logging.set_verbosity_error() # transformers.logging.set_verbosity_error()
warnings.filterwarnings("ignore") warnings.filterwarnings("ignore")
logging.disable(logging.WARNING) logging.disable(logging.WARNING)