dnrops
/
candle
mirror of https://github.com/huggingface/candle.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Upgrading hf-hub.

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
Show Stats 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 (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 {