summaryrefslogtreecommitdiff
path: root/candle-examples/examples/musicgen/t5_model.rs
Commit message (Expand)AuthorAgeFilesLines
* Extract T5 module and add main function to use it (#829)Juarez Bochi2023-09-131-434/+0
* Proper log buckets for t5. (#727)Laurent Mazare2023-09-031-37/+74
* Musicgen text embeddings. (#726)Laurent Mazare2023-09-031-9/+50
* Fix the musicgen example. (#724)Laurent Mazare2023-09-031-31/+15
* Add a simple Module trait and implement it for the various nn layers (#500)Laurent Mazare2023-08-181-0/+1
* Softmax numerical stability. (#267)Laurent Mazare2023-07-281-1/+1
* Rename the .r functions to .dims so as to be a bit more explicit. (#220)Laurent Mazare2023-07-221-1/+1
* Simplify the parameters used by sum and sum_keepdim. (#165)Laurent Mazare2023-07-141-1/+1
* Use the same default as pytorch for sum. (#164)Laurent Mazare2023-07-131-1/+1
* Encodec forward pass (#153)Laurent Mazare2023-07-131-2/+3
* Add the forward pass for the T5 model. (#152)Laurent Mazare2023-07-121-4/+102
* MusicGen var-store path cleanup. (#132)Laurent Mazare2023-07-101-47/+31
* [nn] Move the Embedding and Activation parts. (#116)Laurent Mazare2023-07-101-3/+3
* Sketch the candle-nn crate. (#115)Laurent Mazare2023-07-101-7/+7
* Sketching the musicgen model. (#66)Laurent Mazare2023-07-091-0/+288