diff options
| -rw-r--r-- | candle-examples/examples/bert/main.rs | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/candle-examples/examples/bert/main.rs b/candle-examples/examples/bert/main.rs index b0c0e18a..6cee66ee 100644 --- a/candle-examples/examples/bert/main.rs +++ b/candle-examples/examples/bert/main.rs @@ -62,16 +62,16 @@ impl Args { let repo = Repo::with_revision(model_id, RepoType::Model, revision); let (config_filename, tokenizer_filename, weights_filename) = if self.offline { - let cache = Cache::default(); + let cache = Cache::default().repo(repo); ( cache - .get(&repo, "config.json") + .get("config.json") .ok_or(anyhow!("Missing config file in cache"))?, cache - .get(&repo, "tokenizer.json") + .get("tokenizer.json") .ok_or(anyhow!("Missing tokenizer file in cache"))?, cache - .get(&repo, "model.safetensors") + .get("model.safetensors") .ok_or(anyhow!("Missing weights file in cache"))?, ) } else { |