huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Upgrading hf-hub.

Browse Source
This commit is contained in:
Nicolas Patry
2023-08-29 14:18:54 +02:00
parent 4ed202447e
commit 1aca6fa291
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
candle-examples/examples/bert/main.rs
View File

@ -62,16 +62,16 @@ impl Args {
let repo = Repo::with_revision(model_id, RepoType::Model, revision); let repo = Repo::with_revision(model_id, RepoType::Model, revision);
let (config_filename, tokenizer_filename, weights_filename) = if self.offline { let (config_filename, tokenizer_filename, weights_filename) = if self.offline {
let cache = Cache::default(); let cache = Cache::default().repo(repo);
( (
cache cache
.get(&repo, "config.json") .get("config.json")
.ok_or(anyhow!("Missing config file in cache"))?, .ok_or(anyhow!("Missing config file in cache"))?,
cache cache
.get(&repo, "tokenizer.json") .get("tokenizer.json")
.ok_or(anyhow!("Missing tokenizer file in cache"))?, .ok_or(anyhow!("Missing tokenizer file in cache"))?,
cache cache
.get(&repo, "model.safetensors") .get("model.safetensors")
.ok_or(anyhow!("Missing weights file in cache"))?, .ok_or(anyhow!("Missing weights file in cache"))?,
) )
} else { } else {