Kohaku-Blueleaf bbcc65e7e0

Remove grad ckpt from model impl ...

we apply grad ckpt in train node directly

2025-06-13 17:51:37 +08:00

..

cosmos_tokenizer

Lower cosmos VAE memory usage by a bit.

2025-01-15 22:57:52 -05:00

blocks.py

Remove grad ckpt from model impl

2025-06-13 17:51:37 +08:00

model.py

Change cosmos and hydit models to use the native RMSNorm. (#7934)

2025-05-04 06:26:20 -04:00

position_embedding.py

Fix some cosmos fp8 issues.

2025-01-16 17:45:37 -05:00

vae.py

Fix cosmos VAE failing with videos longer than 121 frames.

2025-01-16 16:30:06 -05:00