diff options
Diffstat (limited to 'candle-examples/examples/falcon/main.rs')
| -rw-r--r-- | candle-examples/examples/falcon/main.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/falcon/main.rs b/candle-examples/examples/falcon/main.rs index 84de6480..a59a0349 100644 --- a/candle-examples/examples/falcon/main.rs +++ b/candle-examples/examples/falcon/main.rs @@ -169,7 +169,7 @@ fn main() -> Result<()> { let vb = VarBuilder::from_safetensors(weights, DTYPE, &device); let config = Config::falcon7b(); config.validate()?; - let model = Falcon::load(&vb, config)?; + let model = Falcon::load(vb, config)?; println!("loaded the model in {:?}", start.elapsed()); let mut pipeline = TextGeneration::new(model, tokenizer, args.seed, args.temperature, &device); |