mirror of
https://github.com/huggingface/candle.git
synced 2025-06-17 19:18:50 +00:00
01545f7303
* Add a slice_set op. * Add some testing. * Add the dedicated kv-cache module. * Derive debug and clone. * Expose more kv-cache functions. * Return the current data when appending. * Use the new cache in the quantized phi3 model.
42 lines
1.3 KiB
Rust
42 lines
1.3 KiB
Rust
pub mod activation;
|
|
pub mod batch_norm;
|
|
pub mod conv;
|
|
pub mod embedding;
|
|
pub mod encoding;
|
|
pub mod func;
|
|
pub mod group_norm;
|
|
pub mod init;
|
|
pub mod kv_cache;
|
|
pub mod layer_norm;
|
|
pub mod linear;
|
|
pub mod loss;
|
|
pub mod ops;
|
|
pub mod optim;
|
|
pub mod rnn;
|
|
pub mod rotary_emb;
|
|
pub mod sequential;
|
|
pub mod var_builder;
|
|
pub mod var_map;
|
|
|
|
pub use activation::{prelu, Activation, PReLU};
|
|
pub use batch_norm::{batch_norm, BatchNorm, BatchNormConfig};
|
|
pub use conv::{
|
|
conv1d, conv1d_no_bias, conv2d, conv2d_no_bias, conv_transpose1d, conv_transpose1d_no_bias,
|
|
conv_transpose2d, conv_transpose2d_no_bias, Conv1d, Conv1dConfig, Conv2d, Conv2dConfig,
|
|
ConvTranspose1d, ConvTranspose1dConfig, ConvTranspose2d, ConvTranspose2dConfig,
|
|
};
|
|
pub use embedding::{embedding, Embedding};
|
|
pub use func::{func, func_t, Func, FuncT};
|
|
pub use group_norm::{group_norm, GroupNorm};
|
|
pub use init::Init;
|
|
pub use layer_norm::{layer_norm, rms_norm, LayerNorm, LayerNormConfig, RmsNorm};
|
|
pub use linear::{linear, linear_b, linear_no_bias, Linear};
|
|
pub use ops::Dropout;
|
|
pub use optim::{AdamW, Optimizer, ParamsAdamW, SGD};
|
|
pub use rnn::{gru, lstm, GRUConfig, LSTMConfig, GRU, LSTM, RNN};
|
|
pub use sequential::{seq, Sequential};
|
|
pub use var_builder::VarBuilder;
|
|
pub use var_map::VarMap;
|
|
|
|
pub use candle::{Module, ModuleT};
|