Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
ebbba62c
Unverified
Commit
ebbba62c
authored
Jun 16, 2022
by
Patrick von Platen
Committed by
GitHub
Jun 16, 2022
Browse files
Merge pull request #18 from vvvm23/logging-transformers-to-diffusers
changes comments and env vars in `utils/logging.py`
parents
92665866
c3cc8eb2
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
15 additions
and
15 deletions
+15
-15
src/diffusers/utils/logging.py
src/diffusers/utils/logging.py
+15
-15
No files found.
src/diffusers/utils/logging.py
View file @
ebbba62c
...
@@ -49,16 +49,16 @@ _tqdm_active = True
...
@@ -49,16 +49,16 @@ _tqdm_active = True
def
_get_default_logging_level
():
def
_get_default_logging_level
():
"""
"""
If
TRANSFORM
ERS_VERBOSITY env var is set to one of the valid choices return that as the new default level. If it is
If
DIFFUS
ERS_VERBOSITY env var is set to one of the valid choices return that as the new default level. If it is
not - fall back to `_default_log_level`
not - fall back to `_default_log_level`
"""
"""
env_level_str
=
os
.
getenv
(
"
TRANSFORM
ERS_VERBOSITY"
,
None
)
env_level_str
=
os
.
getenv
(
"
DIFFUS
ERS_VERBOSITY"
,
None
)
if
env_level_str
:
if
env_level_str
:
if
env_level_str
in
log_levels
:
if
env_level_str
in
log_levels
:
return
log_levels
[
env_level_str
]
return
log_levels
[
env_level_str
]
else
:
else
:
logging
.
getLogger
().
warning
(
logging
.
getLogger
().
warning
(
f
"Unknown option
TRANSFORM
ERS_VERBOSITY=
{
env_level_str
}
, "
f
"Unknown option
DIFFUS
ERS_VERBOSITY=
{
env_level_str
}
, "
f
"has to be one of:
{
', '
.
join
(
log_levels
.
keys
())
}
"
f
"has to be one of:
{
', '
.
join
(
log_levels
.
keys
())
}
"
)
)
return
_default_log_level
return
_default_log_level
...
@@ -126,14 +126,14 @@ def get_logger(name: Optional[str] = None) -> logging.Logger:
...
@@ -126,14 +126,14 @@ def get_logger(name: Optional[str] = None) -> logging.Logger:
def
get_verbosity
()
->
int
:
def
get_verbosity
()
->
int
:
"""
"""
Return the current level for the 🤗
Transform
ers'
s
root logger as an int.
Return the current level for the 🤗
Diffus
ers' root logger as an int.
Returns:
Returns:
`int`: The logging level.
`int`: The logging level.
<Tip>
<Tip>
🤗
Transform
ers has following logging levels:
🤗
Diffus
ers has following logging levels:
- 50: `diffusers.logging.CRITICAL` or `diffusers.logging.FATAL`
- 50: `diffusers.logging.CRITICAL` or `diffusers.logging.FATAL`
- 40: `diffusers.logging.ERROR`
- 40: `diffusers.logging.ERROR`
...
@@ -149,7 +149,7 @@ def get_verbosity() -> int:
...
@@ -149,7 +149,7 @@ def get_verbosity() -> int:
def
set_verbosity
(
verbosity
:
int
)
->
None
:
def
set_verbosity
(
verbosity
:
int
)
->
None
:
"""
"""
Set the verbosity level for the 🤗
Transform
ers'
s
root logger.
Set the verbosity level for the 🤗
Diffus
ers' root logger.
Args:
Args:
verbosity (`int`):
verbosity (`int`):
...
@@ -187,7 +187,7 @@ def set_verbosity_error():
...
@@ -187,7 +187,7 @@ def set_verbosity_error():
def
disable_default_handler
()
->
None
:
def
disable_default_handler
()
->
None
:
"""Disable the default handler of the HuggingFace
Transform
ers'
s
root logger."""
"""Disable the default handler of the HuggingFace
Diffus
ers' root logger."""
_configure_library_root_logger
()
_configure_library_root_logger
()
...
@@ -196,7 +196,7 @@ def disable_default_handler() -> None:
...
@@ -196,7 +196,7 @@ def disable_default_handler() -> None:
def
enable_default_handler
()
->
None
:
def
enable_default_handler
()
->
None
:
"""Enable the default handler of the HuggingFace
Transform
ers'
s
root logger."""
"""Enable the default handler of the HuggingFace
Diffus
ers' root logger."""
_configure_library_root_logger
()
_configure_library_root_logger
()
...
@@ -205,7 +205,7 @@ def enable_default_handler() -> None:
...
@@ -205,7 +205,7 @@ def enable_default_handler() -> None:
def
add_handler
(
handler
:
logging
.
Handler
)
->
None
:
def
add_handler
(
handler
:
logging
.
Handler
)
->
None
:
"""adds a handler to the HuggingFace
Transform
ers'
s
root logger."""
"""adds a handler to the HuggingFace
Diffus
ers' root logger."""
_configure_library_root_logger
()
_configure_library_root_logger
()
...
@@ -214,7 +214,7 @@ def add_handler(handler: logging.Handler) -> None:
...
@@ -214,7 +214,7 @@ def add_handler(handler: logging.Handler) -> None:
def
remove_handler
(
handler
:
logging
.
Handler
)
->
None
:
def
remove_handler
(
handler
:
logging
.
Handler
)
->
None
:
"""removes given handler from the HuggingFace
Transform
ers'
s
root logger."""
"""removes given handler from the HuggingFace
Diffus
ers' root logger."""
_configure_library_root_logger
()
_configure_library_root_logger
()
...
@@ -233,7 +233,7 @@ def disable_propagation() -> None:
...
@@ -233,7 +233,7 @@ def disable_propagation() -> None:
def
enable_propagation
()
->
None
:
def
enable_propagation
()
->
None
:
"""
"""
Enable propagation of the library log outputs. Please disable the HuggingFace
Transform
ers'
s
default handler to
Enable propagation of the library log outputs. Please disable the HuggingFace
Diffus
ers' default handler to
prevent double logging if the root logger has been configured.
prevent double logging if the root logger has been configured.
"""
"""
...
@@ -243,7 +243,7 @@ def enable_propagation() -> None:
...
@@ -243,7 +243,7 @@ def enable_propagation() -> None:
def
enable_explicit_format
()
->
None
:
def
enable_explicit_format
()
->
None
:
"""
"""
Enable explicit formatting for every HuggingFace
Transform
ers'
s
logger. The explicit formatter is as follows:
Enable explicit formatting for every HuggingFace
Diffus
ers' logger. The explicit formatter is as follows:
```
```
[LEVELNAME|FILENAME|LINE NUMBER] TIME >> MESSAGE
[LEVELNAME|FILENAME|LINE NUMBER] TIME >> MESSAGE
```
```
...
@@ -258,7 +258,7 @@ def enable_explicit_format() -> None:
...
@@ -258,7 +258,7 @@ def enable_explicit_format() -> None:
def
reset_format
()
->
None
:
def
reset_format
()
->
None
:
"""
"""
Resets the formatting for HuggingFace
Transform
ers'
s
loggers.
Resets the formatting for HuggingFace
Diffus
ers' loggers.
All handlers currently bound to the root logger are affected by this method.
All handlers currently bound to the root logger are affected by this method.
"""
"""
...
@@ -270,10 +270,10 @@ def reset_format() -> None:
...
@@ -270,10 +270,10 @@ def reset_format() -> None:
def
warning_advice
(
self
,
*
args
,
**
kwargs
):
def
warning_advice
(
self
,
*
args
,
**
kwargs
):
"""
"""
This method is identical to `logger.warninging()`, but if env var
TRANSFORM
ERS_NO_ADVISORY_WARNINGS=1 is set, this
This method is identical to `logger.warninging()`, but if env var
DIFFUS
ERS_NO_ADVISORY_WARNINGS=1 is set, this
warning will not be printed
warning will not be printed
"""
"""
no_advisory_warnings
=
os
.
getenv
(
"
TRANSFORM
ERS_NO_ADVISORY_WARNINGS"
,
False
)
no_advisory_warnings
=
os
.
getenv
(
"
DIFFUS
ERS_NO_ADVISORY_WARNINGS"
,
False
)
if
no_advisory_warnings
:
if
no_advisory_warnings
:
return
return
self
.
warning
(
*
args
,
**
kwargs
)
self
.
warning
(
*
args
,
**
kwargs
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment