Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

multiple small stylistic changes requested by reviewers

Browse Source
This commit is contained in:
Lincoln Stein
2023-11-08 16:45:26 -05:00
parent ce22c0fbaa
commit 6b173cc66f
5 changed files with 110 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/backend/util/util.py
View File

@ -386,13 +386,10 @@ class Chdir(object):
class SilenceWarnings(object):
"""Context manager to temporarily lower verbosity of diffusers & transformers warning messages."""
def __init__(self):
"""Set up context, save current transformers and diffusers verbosity settings."""
self.transformers_verbosity = transformers_logging.get_verbosity()
self.diffusers_verbosity = diffusers_logging.get_verbosity()
def __enter__(self):
"""Set verbosity to error."""
self.transformers_verbosity = transformers_logging.get_verbosity()
self.diffusers_verbosity = diffusers_logging.get_verbosity()
transformers_logging.set_verbosity_error()
diffusers_logging.set_verbosity_error()
warnings.simplefilter("ignore")