huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixing falcon example.

Browse Source
This commit is contained in:
Nicolas Patry
2023-07-07 00:29:44 +02:00
parent 514b171f75
commit 1ec221a749
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
candle-examples/examples/falcon/main.rs
View File

@ -117,6 +117,7 @@ struct Args {
fn main() -> Result<()> { fn main() -> Result<()> {
use candle_hub::{api::sync::Api, Repo, RepoType}; use candle_hub::{api::sync::Api, Repo, RepoType};
use tokenizers::Tokenizer;
let args = Args::parse(); let args = Args::parse();
let device = if args.cpu { let device = if args.cpu {