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