diff options
Diffstat (limited to 'candle-transformers/src/models/gemma2.rs')
| -rw-r--r-- | candle-transformers/src/models/gemma2.rs | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/candle-transformers/src/models/gemma2.rs b/candle-transformers/src/models/gemma2.rs index f0d65047..ec23efc5 100644 --- a/candle-transformers/src/models/gemma2.rs +++ b/candle-transformers/src/models/gemma2.rs @@ -1,3 +1,9 @@ +//! Gemma LLM architecture (Google) inference implementation. +//! +//! See ["Gemma: Open Models Based on Gemini Technology"](https://blog.google/technology/developers/gemma-open-models/) +//! +//! Based on implementations from Google and OpenLLM + use std::sync::Arc; use candle::{DType, Device, Module, Result, Tensor, D}; |