summaryrefslogtreecommitdiff
path: root/candle-examples/examples/falcon/main.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-examples/examples/falcon/main.rs')
-rw-r--r--candle-examples/examples/falcon/main.rs3
1 files changed, 1 insertions, 2 deletions
diff --git a/candle-examples/examples/falcon/main.rs b/candle-examples/examples/falcon/main.rs
index 05507f08..c45fe545 100644
--- a/candle-examples/examples/falcon/main.rs
+++ b/candle-examples/examples/falcon/main.rs
@@ -14,8 +14,7 @@ use clap::Parser;
use hf_hub::{api::sync::Api, Repo, RepoType};
use tokenizers::Tokenizer;
-mod model;
-use model::{Config, Falcon};
+use candle_transformers::models::falcon::{Config, Falcon};
struct TextGeneration {
model: Falcon,