Commit Graph

23 Commits

Author SHA1 Message Date
Laurent Mazare c58c5d5b01
Add the mimi audio-tokenizer. (#2488)
* Add the mimi audio-tokenizer.

* Formatting tweaks.

* Add a full example.

* Use the transformers names.

* More renamings.

* Get encoding and decoding to work.

* Clippy fixes.
2024-09-20 14:31:20 -06:00
Laurent Mazare 236b29ff15
Add the DAC model. (#2433)
* Add the DAC model.

* More quantization support.

* Handle DAC decoding.

* Plug the DAC decoding in parler-tts.
2024-08-19 08:59:51 +02:00
Laurent Mazare 58197e1896
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.
2024-08-18 20:42:08 +02:00
Laurent Mazare b75ef051cf
Fix the marian tokenizer importer. (#2426)
* Fix the marian tokenizer importer.

* Ignore the python caches.
2024-08-17 20:58:40 +02:00
shua 9182f9f5c2
ignore editor config folders (#2315) 2024-07-07 19:43:48 +02:00
Radamés Ajna 27e70a5093
Whisper quantized wasm (#1028)
* [Whisper] Update to use quantized model

* [whisper] add language detection

* [whisper] change assets location

* [whisper] adapt js example with quantized models

* [whisper] better task parsing

* [whisper] minor fixes
2023-10-04 20:22:57 +01:00
Laurent Mazare 2619c4307f
Add a quantized version of the t5 model. (#921) 2023-09-21 11:13:39 +01:00
Lei 49326fb925
Update .gitignore (#649) 2023-08-29 08:41:33 +01:00
Radamés Ajna 864227edbf
[WIP] Improve Yolo WASM UI example (#591)
* return detections with classes names

* ignore .DS_Store

* example how to load wasm module

* add param to set model size

* add param for model size

* accept iou and confidence threshold on run

* conf and iou thresholds

* clamp only

* remove images from branch

* a couple of renamings, add readme with instructions

* final design

* minor font + border update
2023-08-26 11:40:41 +01:00
Laurent Mazare 7687a0f453
Also fix the aspect ratio in the wasm example. (#556)
* Also fix the aspect ratio in the wasm example.

* Add the yolo lib.

* Update the build script.
2023-08-22 22:20:08 +01:00
Laurent Mazare 20ce3e9f39
Sketch the yolo wasm example. (#546)
* Sketch the yolo wasm example.

* Web ui.

* Get the web ui to work.

* UI tweaks.

* More UI tweaks.

* Use the natural width/height.

* Add a link to the hf space in the readme.
2023-08-22 11:56:43 +01:00
Laurent Mazare 0b175fcbbd
Fix the pyo3 build for macos. (#324)
* Fix the pyo3 build for macos.

* rustfmt fix.
2023-08-05 14:53:57 +01:00
Laurent Mazare 5a26cba733
Re-organize the wasm examples (#231)
* Move the whisper example.

* More renaming.

* Add llama2 as a new wasm example.

* Live generation.

* More of the llama wasm example.

* Formatting.
2023-07-24 12:36:02 +01:00
Laurent Mazare cb687b4897
Add some more developed training examples. (#199)
* Use contiguous tensors for variables.

* Sketch the mnist example.

* Start adding the reduce ops.

* Renaming.

* Refactor the reduce operations.

* Bugfix for the broadcasting vectorization.
2023-07-19 15:37:52 +01:00
Laurent Mazare f0cccd08f0
Bert tracing (#184)
* Add some tracing to bert.

* More tracing.

* Add a flag for tracing.
2023-07-17 19:40:42 +01:00
Laurent Mazare 2ddda706bd
Switch to using trunk. (#171) 2023-07-14 22:06:40 +01:00
Laurent Mazare d1f6fad84a
Whisper example in wasm. (#169)
* Whisper example in wasm.

* Load the model.

* Get the whisper demo to work (though slowly).

* Polish the UI a bit.

* Minor tweak.

* More UI.

* Add the progress bar.
2023-07-14 19:33:36 +01:00
Laurent Mazare 88f666781f
Wasm proof of concept. (#167)
* Wasm proof of concept.

* Run whisper inference in the browser.

* Some fixes.

* Move the wasm example.

* Change the tokenizer config.
2023-07-14 14:51:46 +01:00
Laurent Mazare ba35d895e7
Sketch the candle-transformers crate. (#147)
* Sketch the candle-transformers crate.

* Format the empty files.
2023-07-12 13:49:31 +01:00
laurent ebb0fedf14 Very simple pyo3 bindings for candle. 2023-07-01 20:36:44 +01:00
laurent 19cbbc5212 Improve how we check that the dims are in bounds. 2023-06-30 09:11:00 +01:00
laurent 90c140ff4b Start sketching the llama example. 2023-06-25 13:51:20 +01:00
Laurent Mazare fe7a5f53a1
Initial commit 2023-06-19 17:06:31 +01:00