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.rs4
1 files changed, 1 insertions, 3 deletions
diff --git a/candle-examples/examples/falcon/main.rs b/candle-examples/examples/falcon/main.rs
index 88dc3800..18adc1e9 100644
--- a/candle-examples/examples/falcon/main.rs
+++ b/candle-examples/examples/falcon/main.rs
@@ -6,6 +6,7 @@ extern crate intel_mkl_src;
use anyhow::{Error as E, Result};
use candle::{DType, Device, Tensor, D};
+use candle_hub::{api::sync::Api, Repo, RepoType};
use clap::Parser;
use rand::{distributions::Distribution, SeedableRng};
use tokenizers::Tokenizer;
@@ -116,9 +117,6 @@ struct Args {
}
fn main() -> Result<()> {
- use candle_hub::{api::sync::Api, Repo, RepoType};
- use tokenizers::Tokenizer;
-
let args = Args::parse();
let device = if args.cpu {
Device::Cpu