diff options
Diffstat (limited to 'candle-transformers/src/models/codegeex4_9b.rs')
| -rw-r--r-- | candle-transformers/src/models/codegeex4_9b.rs | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/candle-transformers/src/models/codegeex4_9b.rs b/candle-transformers/src/models/codegeex4_9b.rs index aaa99fd9..baf47459 100644 --- a/candle-transformers/src/models/codegeex4_9b.rs +++ b/candle-transformers/src/models/codegeex4_9b.rs @@ -1,3 +1,10 @@ +//! CodeGeeX4 - A multi-language code generation model +//! +//! See "CodeGeeX: A Pre-Trained Model For Code Generation with Multilingual Evaluations on HumanEval-X", Qian et al. 2023 +//! - [Arxiv](https://arxiv.org/abs/2303.17568) +//! - [Github](https://github.com/THUDM/CodeGeeX) +//! + use crate::models::with_tracing::{linear_b as linear, Linear}; use candle::{DType, Device, IndexOp, Module, Result, Tensor, D}; use candle_nn::VarBuilder; |