summaryrefslogtreecommitdiff
path: root/candle-examples/examples/bigcode/main.rs
diff options
context:
space:
mode:
authorLaurent Mazare <laurent.mazare@gmail.com>2023-09-10 09:40:27 +0100
committerGitHub <noreply@github.com>2023-09-10 09:40:27 +0100
commitd3f05eae8c4f2df186b46e433be101ac39fceca5 (patch)
tree6ffc43595caec3007fe28efd3bafc7acbdde6e94 /candle-examples/examples/bigcode/main.rs
parent258ac32c3868d4103e90df19af99a3e13c805c4e (diff)
downloadcandle-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.rs3
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;