Jedrzej Kosinski 8fe2dea297 Made CosmosVideo work with optimized_attention_override

2025-08-28 21:23:03 -07:00

..

cosmos_tokenizer

Replace manual x * sigmoid(x) with torch silu in VAE nonlinearity (#9057)

2025-07-30 19:25:56 -04:00

blocks.py

Made CosmosVideo work with optimized_attention_override

2025-08-28 21:23:03 -07:00

model.py

Made CosmosVideo work with optimized_attention_override

2025-08-28 21:23:03 -07:00

position_embedding.py

Uncap cosmos predict2 res and fix mem estimation. (#8518)

2025-06-13 07:30:18 -04:00

predict2.py

Make CosmosPredict2 work with optimized_attention_override

2025-08-28 21:18:34 -07:00

vae.py

Fix cosmos VAE failing with videos longer than 121 frames.

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