Add the CSM model. (#2862)

* Add the CSM model.

* Add some code to load the model.

* Load the text tokenizer.

* Add frame generation.

* Get the sampling to work.

* Rope fix.

* Autoregressive generation.

* Generate some audio file.

* Use the actual prompt.

* Support multiple turns.

* Add a very barebone readme.

* Move some of the shared bits to the model.
This commit is contained in:
Laurent Mazare
2025-04-04 06:48:03 +02:00
committed by GitHub
parent 9d31361c4f
commit cf9d7bf24c
4 changed files with 791 additions and 0 deletions

View File

@ -27,6 +27,7 @@ pub mod codegeex4_9b;
pub mod colpali;
pub mod convmixer;
pub mod convnext;
pub mod csm;
pub mod dac;
pub mod debertav2;
pub mod deepseek2;