Commit Graph

5 Commits

Author SHA1 Message Date
Maxime Ellerbach 4b6fa7d491 re-parenting of some layers to enable proper zero-3 FSDP 2026-06-18 12:44:00 +00:00
Maxime Ellerbach 56a82e578d preparing for training adding some temporary debug code aswell to visualize model output 2026-06-18 12:43:59 +00:00
Maxime Ellerbach 807e364132 big refactor to use models from diffusers and transformers 2026-06-18 12:43:58 +00:00
ZibinDong dfc0170b4d Add FastWAM policy review updates 2026-06-18 12:43:57 +00:00
ZibinDong a343ed3a63 Add FastWAM policy 2026-06-18 12:43:56 +00:00