diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-02-09 15:02:49 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-02-09 15:02:49 +0100 |
| commit | 5657e596cd91b2c93d395459cbbfc6c64cb28c2d (patch) | |
| tree | 05a8976e7a5a84bd3832d2c23a30844f7a780fd7 /candle-transformers/src/models/mod.rs | |
| parent | 0dee8ea19b56018f995f1732cd9e91065b493a29 (diff) | |
| download | candle-5657e596cd91b2c93d395459cbbfc6c64cb28c2d.tar.gz candle-5657e596cd91b2c93d395459cbbfc6c64cb28c2d.tar.bz2 candle-5657e596cd91b2c93d395459cbbfc6c64cb28c2d.zip | |
Add the Qwen2 model (#1684)
* Initial check-in for the qwen2 model.
* More qwen2 inference.
* Polish the qwen example.
* Fix the rope basis.
* Get the inference to work.
* Support different model sizes.
Diffstat (limited to 'candle-transformers/src/models/mod.rs')
| -rw-r--r-- | candle-transformers/src/models/mod.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-transformers/src/models/mod.rs b/candle-transformers/src/models/mod.rs index 810c252c..f3782fff 100644 --- a/candle-transformers/src/models/mod.rs +++ b/candle-transformers/src/models/mod.rs @@ -30,6 +30,7 @@ pub mod quantized_mixformer; pub mod quantized_mpt; pub mod quantized_stable_lm; pub mod quantized_t5; +pub mod qwen2; pub mod repvgg; pub mod resnet; pub mod segment_anything; |