diff options
author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-09-10 09:40:27 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-09-10 09:40:27 +0100 |
commit | d3f05eae8c4f2df186b46e433be101ac39fceca5 (patch) | |
tree | 6ffc43595caec3007fe28efd3bafc7acbdde6e94 /candle-examples/examples/bigcode/main.rs | |
parent | 258ac32c3868d4103e90df19af99a3e13c805c4e (diff) | |
download | candle-d3f05eae8c4f2df186b46e433be101ac39fceca5.tar.gz candle-d3f05eae8c4f2df186b46e433be101ac39fceca5.tar.bz2 candle-d3f05eae8c4f2df186b46e433be101ac39fceca5.zip |
Move some models to candle-transformers so that it's easier to re-use. (#794)
* Move some models to candle-transformers so that they can be shared.
* Also move falcon.
* Move Llama.
* Move whisper (partial).
Diffstat (limited to 'candle-examples/examples/bigcode/main.rs')
-rw-r--r-- | candle-examples/examples/bigcode/main.rs | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/candle-examples/examples/bigcode/main.rs b/candle-examples/examples/bigcode/main.rs index 652cd47f..3540f75d 100644 --- a/candle-examples/examples/bigcode/main.rs +++ b/candle-examples/examples/bigcode/main.rs @@ -7,8 +7,7 @@ extern crate accelerate_src; use anyhow::{Error as E, Result}; use clap::Parser; -mod model; -use model::{Config, GPTBigCode}; +use candle_transformers::models::bigcode::{Config, GPTBigCode}; use candle::{DType, Device, Tensor}; use candle_nn::VarBuilder; |