diff options
Diffstat (limited to 'candle-transformers/src/models/llama2_c.rs')
| -rw-r--r-- | candle-transformers/src/models/llama2_c.rs | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/candle-transformers/src/models/llama2_c.rs b/candle-transformers/src/models/llama2_c.rs index d825d8e4..930c8b8a 100644 --- a/candle-transformers/src/models/llama2_c.rs +++ b/candle-transformers/src/models/llama2_c.rs @@ -2,7 +2,9 @@ //! //! See ["LLaMA 2: Open Foundation and Fine-Tuned Chat Models"](https://arxiv.org/abs/2307.09288) //! -//! Based on the [llama2.c](https://github.com/karpathy/llama2.c) implementation +//! - ⚡ [Interactive Wasm Example](https://huggingface.co/spaces/lmz/candle-llama2) +//! - 💻 llama2.c [GH Link](https://github.com/karpathy/llama2.c) +//! use candle::{DType, Device, IndexOp, Result, Tensor, D}; use candle_nn::linear_no_bias as linear; |