Skip to content

日志记录

🤗 Diffusers 有一个集中的日志系统,可以轻松管理库的详细程度。默认的详细程度设置为 WARNING

要更改详细程度级别,请使用其中一个直接设置器。例如,要将详细程度更改为 INFO 级别。

python
import diffusers

diffusers.logging.set_verbosity_info()

你还可以使用环境变量 DIFFUSERS_VERBOSITY 来覆盖默认的详细程度。你可以将其设置为以下之一:debuginfowarningerrorcritical。例如:

bash
DIFFUSERS_VERBOSITY=error ./myprogram.py

此外,可以通过将环境变量 DIFFUSERS_NO_ADVISORY_WARNINGS 设置为真值(如 1)来禁用某些 warnings。这将禁用由 [logger.warning_advice] 记录的任何警告。例如:

bash
DIFFUSERS_NO_ADVISORY_WARNINGS=1 ./myprogram.py

以下是如何在你的模块或脚本中使用与库相同的日志记录器的示例:

python
from diffusers.utils import logging

logging.set_verbosity_info()
logger = logging.get_logger("diffusers")
logger.info("INFO")
logger.warning("WARN")

logging 模块的所有方法如下所述。主要方法包括 [logging.get_verbosity] 用于获取日志记录器中的当前详细级别,以及 [logging.set_verbosity] 用于将详细级别设置为你选择的级别。

从最不详细到最详细的顺序如下:

MethodInteger valueDescription
diffusers.logging.CRITICAL or diffusers.logging.FATAL50only report the most critical errors
diffusers.logging.ERROR40only report errors
diffusers.logging.WARNING or diffusers.logging.WARN30only report errors and warnings (default)
diffusers.logging.INFO20only report errors, warnings, and basic information
diffusers.logging.DEBUG10report all information

默认情况下,在模型下载期间会显示 tqdm 进度条。使用 [logging.disable_progress_bar] 和 [logging.enable_progress_bar] 来启用或禁用此行为。

基础设置器

[[autodoc]] utils.logging.set_verbosity_error

[[autodoc]] utils.logging.set_verbosity_warning

[[autodoc]] utils.logging.set_verbosity_info

[[autodoc]] utils.logging.set_verbosity_debug

其他函数

[[autodoc]] utils.logging.get_verbosity

[[autodoc]] utils.logging.set_verbosity

[[autodoc]] utils.logging.get_logger

[[autodoc]] utils.logging.enable_default_handler

[[autodoc]] utils.logging.disable_default_handler

[[autodoc]] utils.logging.enable_explicit_format

[[autodoc]] utils.logging.reset_format

[[autodoc]] utils.logging.enable_progress_bar

[[autodoc]] utils.logging.disable_progress_bar