Commit Graph

6 Commits

Author SHA1 Message Date
Maxime Ellerbach 2f12f58e33 adding lazy imports 2026-06-22 14:26:59 +00:00
Maxime Ellerbach e4ab2feb17 re-parenting of some layers to enable proper zero-3 FSDP 2026-06-22 14:26:54 +00:00
Maxime Ellerbach 4438b238ba preparing for training adding some temporary debug code aswell to visualize model output 2026-06-22 14:26:53 +00:00
Maxime Ellerbach c9a7379b6b big refactor to use models from diffusers and transformers 2026-06-22 14:26:52 +00:00
ZibinDong 27951a42f8 Add FastWAM policy review updates 2026-06-22 14:26:51 +00:00
ZibinDong 7f1717550d Add FastWAM policy 2026-06-22 14:26:50 +00:00