diff options
Diffstat (limited to 'candle-nn/src/lib.rs')
| -rw-r--r-- | candle-nn/src/lib.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-nn/src/lib.rs b/candle-nn/src/lib.rs index 5c0fbb37..fcac5830 100644 --- a/candle-nn/src/lib.rs +++ b/candle-nn/src/lib.rs @@ -6,6 +6,7 @@ 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; |