|
|
49d3f7f708
|
Add support to flan-t5 (#840)
|
2023-09-13 19:27:20 +02:00 |
|
|
|
3e94324012
|
Add some sentence similarity part to the t5 example. (#835)
* Add some sentence similarity part to the t5 example.
* Clippy fix.
|
2023-09-13 10:44:02 +01:00 |
|
|
|
e4553fb355
|
T5 tweaks (#831)
* Use default values rather than options.
* Avoid exposing the device field.
* More tweaks.
|
2023-09-13 07:37:04 +01:00 |
|
|
|
d801e1d564
|
Clippy fix. (#830)
|
2023-09-13 07:16:20 +01:00 |
|
|
|
9daa6dbe87
|
Extract T5 module and add main function to use it (#829)
* Extract t5 out of musicgen
* Add main for t5 module
|
2023-09-13 07:14:05 +01:00 |
|
|
|
805bf9ffa7
|
Implement top_p / nucleus sampling (#819)
* Implement top_p / nucleus sampling
* Update changelog
* rustfmt
* Add tests
* Fix clippy warning
* Fix another clippy error
|
2023-09-12 18:10:16 +02:00 |
|
|
|
c5a058b169
|
Use the module trait in stable-diffusion. (#817)
|
2023-09-11 20:40:07 +01:00 |
|
|
|
d7b9fec849
|
Move the stable-diffusion modeling code so that it's easier to re-use. (#812)
|
2023-09-11 11:45:57 +01:00 |
|
|
|
84ee870efd
|
Use softmax-last-dim in whisper. (#810)
|
2023-09-11 11:05:05 +01:00 |
|
|
|
90e077e409
|
Return the low res mask in the wasm segment-anything module. (#798)
* Return the low res mask.
* Add some validations.
|
2023-09-10 13:03:02 +01:00 |
|
|
|
584171cae1
|
Add a wasm module for the segment anything example. (#797)
|
2023-09-10 12:29:37 +01:00 |
|
|
|
35f72514f5
|
Move more models to candle-transformers (#796)
* Move dinov2.
* Move efficientnet.
* Move the quantized llama model.
* Move segment-anything.
|
2023-09-10 10:20:18 +01:00 |
|
|
|
d3f05eae8c
|
Move some models to candle-transformers so that it's easier to re-use. (#794)
* Move some models to candle-transformers so that they can be shared.
* Also move falcon.
* Move Llama.
* Move whisper (partial).
|
2023-09-10 09:40:27 +01:00 |
|
|
|
6e485f2deb
|
Add some optional repeat penalty. (#623)
* Add some optional repeat penalty.
* Add the missing files.
|
2023-08-27 10:48:45 +01:00 |
|
|
|
912561614f
|
Better handling of zero temperatures. (#532)
|
2023-08-21 07:51:46 +01:00 |
|
|
|
3eb2bc6d07
|
Softmax numerical stability. (#267)
* Softmax numerical stability.
* Fix the flash-attn test.
|
2023-07-28 13:13:01 +01:00 |
|
|
|
ba35d895e7
|
Sketch the candle-transformers crate. (#147)
* Sketch the candle-transformers crate.
* Format the empty files.
|
2023-07-12 13:49:31 +01:00 |
|
