VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-08 22:35:09 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
1cefb2a753680bf472964f145d039065521fd624
Kohya-ss-sd-scripts/library/train_util.py
Kohaku-Blueleaf 1cefb2a753 Better implementation for te autocast (#895) 
* Better implementation for te

* Fix some misunderstanding

* as same as unet, add explicit convert

* Better cache TE and TE lr

* Fix with list

* Add timeout settings

* Fix arg style
2023-10-28 15:49:59 +09:00

197 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink