Plat a823fd9fb8 Improve wandb logging (#1576) ...

* fix: wrong training steps were recorded to wandb, and no log was sent when logging_dir was not specified

* fix: checking of whether wandb is enabled

* feat: log images to wandb with their positive prompt as captions

* feat: logging sample images' caption for sd3 and flux

* fix: import wandb before use

2024-09-11 22:21:16 +09:00

42 KiB

Raw Blame History

View Raw