1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

pub use classification::*;
pub use conditional_generation::*;
pub use conditional_generation_with_pkvs::*;
pub use embedding::*;
pub use seq2seq_decoder::*;
pub use seq2seq_decoder_with_pkvs::*;
pub use seq2seq_encdec::*;
pub use seq2seq_encoder::*;

pub mod classification;
pub mod conditional_generation;
pub mod conditional_generation_with_pkvs;
pub mod embedding;
pub mod seq2seq_decoder;
pub mod seq2seq_decoder_with_pkvs;
pub mod seq2seq_encdec;
pub mod seq2seq_encoder;

// Doesn't work: see https://github.com/huggingface/transformers/issues/16512
// pub use seq2seq_generation_with_pkvs::*;