Kohya-ss-sd-scripts/library/deepspeed_utils.py at d33d5eccd16970e489359ee02b89a6259559e4b9 - Kohya-ss-sd-scripts - Optimize GIT

VictorMylle/Kohya-ss-sd-scripts

mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-10 15:00:23 +00:00

Files

saibit 7c61c0dfe0 Add autocast warpper for forward functions in deepspeed_utils.py to try aligning precision when using mixed precision in training process

2025-04-22 16:06:55 +08:00

6.5 KiB

Raw Blame History

View Raw