59 Commits

Author SHA1 Message Date
gameloader
50b1f76263 feat(models/denoiser): Add Mamba2 model implementation
Implement a minimal, single-file implementation of the Mamba-2 model in PyTorch.
Based on the paper "Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Duality".
2026-01-20 14:44:03 +08:00
gameloader
b9c3720eba feat(callbacks): add Weights & Biases image logging to SaveImagesHook\n\n- Introduce optional wandb_max_save_num parameter in __init__\n- Add helper methods _get_wandb_logger and _log_wandb_samples\n- Log sampled images as wandb.Image artifacts during validation epoch end (global zero only)\n- Handle proper dtype conversion and clipping for wandb compatibility\n- Respect both general max_save_num and wandb-specific limit\n\nThis enables visualization of validation samples directly in the W&B dashboard without manual intervention. 2026-01-19 16:03:54 +08:00
game-loader
6bf32b08fd feat(dataset): add HuggingFace image dataset 2026-01-18 17:10:01 +08:00
game-loader
cf45afe325 feat(data): add support for Hugging Face datasets 2026-01-16 14:25:32 +08:00
wang shuai
1d1b4d2913 Update README.md 2025-08-22 10:49:18 +08:00
wangshuai6
ae7df43ecc torch.compile 2025-07-03 20:23:59 +08:00
wang shuai
2c16b8f423 Update improved_dit.py 2025-07-03 20:21:04 +08:00
wangshuai6
8038c16bee disperse loss 2025-06-14 12:14:01 +08:00
wangshuai6
598f7b40a2 disperse loss 2025-06-13 10:41:31 +08:00
wangshuai6
3093a65151 disperse loss 2025-06-12 22:18:15 +08:00
wangshuai6
910b764275 pixelddt-t2i update 2025-06-05 09:56:03 +08:00
wangshuai6
26c8e54edb pixelddt-t2i update 2025-06-05 09:55:43 +08:00
wangshuai6
99d92c94e7 fix bugs(admas timedeltas) 2025-05-20 12:30:40 +08:00
wangshuai6
3693640ca3 fix bugs(admas timedeltas) 2025-05-20 12:27:34 +08:00
wangshuai6
699d89a772 Merge remote-tracking branch 'origin/main' 2025-05-20 12:26:14 +08:00
wangshuai6
4d4b3262c3 fix bugs(admas timedeltas) 2025-05-20 12:25:50 +08:00
wang shuai
95efdac7d5 Merge pull request #5 from eltociear/patch-1
docs: update README.md
2025-04-15 17:47:45 +08:00
Ikko Eltociear Ashimine
39a5a47961 docs: update README.md
surpresingly -> surprisingly
2025-04-15 17:27:35 +09:00
wangshuai6
45d5e8db5b fix bugs(update README) 2025-04-15 14:28:07 +08:00
wangshuai6
d02bfe7879 fix bugs(update README) 2025-04-14 09:01:22 +08:00
wangshuai6
1ab6d0d47a fix bugs(update README) 2025-04-14 09:00:03 +08:00
wangshuai6
5c087aa833 fix bugs(update README) 2025-04-14 08:57:53 +08:00
wangshuai6
6990e30ef5 fix bugs(update README) 2025-04-14 08:57:26 +08:00
wangshuai6
c5f0effff9 fix bugs(update README) 2025-04-14 08:57:05 +08:00
wangshuai6
0d7e2e3619 fix bugs(update README) 2025-04-14 08:51:25 +08:00
wangshuai6
2eec7b6479 fix bugs(update README) 2025-04-14 08:45:57 +08:00
wangshuai6
9010659f72 fix bugs(update src) 2025-04-14 08:45:47 +08:00
wangshuai6
ea60d28838 fix bugs(update dependency) 2025-04-14 08:45:31 +08:00
wangshuai6
a307da32e1 fix bugs(update configs) 2025-04-13 09:23:09 +08:00
wangshuai6
ce93cc61a8 fix bugs(rm encoder.py) 2025-04-13 09:14:35 +08:00
wangshuai6
dacb14e478 fix bugs 2025-04-13 09:08:31 +08:00
wangshuai6
b9d6dff987 update acknowledge 2025-04-11 20:59:07 +08:00
wangshuai6
905769dedb update bibtex 2025-04-11 20:51:33 +08:00
wangshuai6
d387955e63 app demo for hf spaces 2025-04-11 19:50:51 +08:00
wangshuai6
28d9af791d app demo 2025-04-11 19:40:27 +08:00
wangshuai6
4591692c30 app demo 2025-04-11 19:31:52 +08:00
wangshuai6
687d650175 app demo 2025-04-11 19:14:51 +08:00
wangshuai6
c1c8043ed1 app demo 2025-04-11 19:10:11 +08:00
wangshuai6
0972ab65b4 app demo 2025-04-11 18:30:34 +08:00
wangshuai6
d961ccc57a app demo 2025-04-11 18:00:29 +08:00
wangshuai6
12227c4f7f app demo 2025-04-11 17:41:00 +08:00
wangshuai6
ed3c466d08 app demo 2025-04-11 16:51:33 +08:00
wangshuai6
609cf377cb fix bugs 2025-04-11 13:24:51 +08:00
wangshuai6
56ff957581 badge 2025-04-11 12:25:58 +08:00
wangshuai6
d8310feb62 badge 2025-04-11 12:25:28 +08:00
wangshuai6
32cc2754bf badge 2025-04-11 12:22:30 +08:00
wangshuai6
485818abf6 remove hardcode dino encoder logic 2025-04-11 11:07:59 +08:00
wangshuai6
093b0f47f8 remove timeshift to align performance 2025-04-11 11:02:22 +08:00
wangshuai6
5c3867ffdc dynamic programming 2025-04-10 10:02:41 +08:00
wangshuai6
b6792002e6 update README 2025-04-09 11:49:03 +08:00