AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

[Chore] switch to logger.warning (#7289)

Browse Source 
switch to logger.warning
This commit is contained in:
Sayak Paul
2024-03-13 06:56:43 +05:30
committed by GitHub
parent 2ea28d69dc
commit 4fbd310fd2
61 changed files with 138 additions and 136 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/custom_diffusion/train_custom_diffusion.py
View File

@@ -904,7 +904,7 @@ def main(args):
xformers_version = version.parse(xformers.__version__)
if xformers_version == version.parse("0.0.16"):
logger.warn(
logger.warning(
"xFormers 0.0.16 cannot be used for training in some GPUs. If you observe problems during training, please update xFormers to at least 0.0.17. See https://huggingface.co/docs/diffusers/main/en/optimization/xformers for more details."
)
attention_class = CustomDiffusionXFormersAttnProcessor