Encodec model. (#1771)

* Encodec model.

* Fixes.

* Add the padding functions.

* Get the LSTM bit to work.

* Get the encodec model to generate some tokens (decoder only for now).

* Minor tweak.

* Minor tweak.
This commit is contained in:
Laurent Mazare
2024-02-27 22:59:40 +01:00
committed by GitHub
parent 205767f9de
commit 0c49e95dfb
7 changed files with 834 additions and 1 deletions

View File

@ -1,6 +1,7 @@
pub mod coco_classes;
pub mod imagenet;
pub mod token_output_stream;
pub mod wav;
use candle::utils::{cuda_is_available, metal_is_available};
use candle::{Device, Result, Tensor};