4 commits

Author	SHA1	Message	Date
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