hooved
|
1e8945a28c
|
Training loop for Stable Diffusion mlperf (#12315)
* add diff
* fix edit error
* match master
* point reference to specific commit
* simplify wandb logging
* remove lr test, dehardcode device
* increase stack size limit
|
2025-10-03 02:45:38 -04:00 |
|
hooved
|
5d9035f5a6
|
Eval for Stable Diffusion mlperf (#12316)
* add diff
* rerun ci
* refactor beam workaround, add test
* fix conflict
* linting
|
2025-10-02 02:35:38 -04:00 |
|
hooved
|
0f804c9a83
|
Stable Diffusion model init for mlperf (#12314)
* include clip pr diff
* updated unet and sd init
* dehardcode default device
* revert beam hang workaround
---------
Co-authored-by: chenyu <chenyu@fastmail.com>
|
2025-10-02 02:28:41 -04:00 |
|
hooved
|
c2689c505e
|
Clip model updates for Stable Diffusion mlperf training (#12313)
* stable diffusion mlperf clip changes
* add clip tests
* set gelu as attribute
* add more tests
* factor out GPUS
* rerun CI
* add imports to if blocks
* remove unneeded axis
* add clip tests to CI
* move clip tests
* add deps, disable max buf size
|
2025-09-29 21:50:14 -04:00 |
|