mirror of
https://github.com/huggingface/candle.git
synced 2025-06-19 11:56:45 +00:00
Documentation Pass for Models (#2617)
* links in chinese_clip * links for clip model * add mod docs for flux and llava * module doc for MMDIT and MIMI * add docs for a few more modesl * mod docs for bert naser and beit * add module docs for convmixer colpali codegeex and chatglm * add another series of moddocs * add fastvit-llama2_c * module docs mamba -> mobileone * module docs from moondream-phi3 * mod docs for quantized and qwen * update to yi * fix long names * Update llama2_c.rs * Update llama2_c_weights.rs * Fix the link for mimi + tweaks --------- Co-authored-by: Laurent Mazare <laurent.mazare@gmail.com>
This commit is contained in:
@ -1,3 +1,20 @@
|
||||
//! Recurrent Gemma model implementation with quantization support.
|
||||
//!
|
||||
//! Gemma is a large language model optimized for efficiency.
|
||||
//! This implementation provides quantization for reduced memory and compute.
|
||||
//!
|
||||
//! Key characteristics:
|
||||
//! - Recurrent blocks with gated recurrent units
|
||||
//! - Convolution and attention blocks
|
||||
//! - RMSNorm for layer normalization
|
||||
//! - Rotary positional embeddings (RoPE)
|
||||
//! - Support for 8-bit quantization
|
||||
//!
|
||||
//! References:
|
||||
//! - [Gemma Paper](https://arxiv.org/abs/2401.06751)
|
||||
//! - [Model Card](https://ai.google.dev/gemma)
|
||||
//!
|
||||
|
||||
use crate::quantized_nn::{linear_b as linear, Embedding, Linear};
|
||||
pub use crate::quantized_var_builder::VarBuilder;
|
||||
use candle::{DType, Device, IndexOp, Module, Result, Tensor, D};
|
||||
|
Reference in New Issue
Block a user