9 Commits

Author	SHA1	Message	Date
Maxime Ellerbach	218f8a1735	moving and renaming files to have a cleaner file tree	2026-06-30 11:01:33 +00:00
Maxime Ellerbach	343ab5f99c	make tokenizer/text-encoder model ids configurable + some nits	2026-06-23 09:07:16 +00:00
Maxime Ellerbach	5f6ddab629	fixing stale assertion	2026-06-22 14:27:01 +00:00
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