parler-tts support (#2431)

* Start sketching parler-tts support.

* Implement the attention.

* Add the example code.

* Fix the example.

* Add the description + t5 encode it.

* More of the parler forward pass.

* Fix the positional embeddings.

* Support random sampling in generation.

* Handle EOS.

* Add the python decoder.

* Proper causality mask.
This commit is contained in:
Laurent Mazare
2024-08-18 19:42:08 +01:00
committed by GitHub
parent 736d8eb752
commit 58197e1896
5 changed files with 658 additions and 0 deletions

View File

@ -40,6 +40,7 @@ pub mod mobileone;
pub mod moondream;
pub mod mpt;
pub mod olmo;
pub mod parler_tts;
pub mod persimmon;
pub mod phi;
pub mod phi3;