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

Merge pull request #658 from huggingface/upgrade_hf_hub2

Browse Source 
Upgrading hf-hub (for windows support, removing symlink requirement).
This commit is contained in:
Nicolas Patry
2023-08-29 14:32:15 +02:00
committed by GitHub
parent 1d6bff53fc 1aca6fa291
commit 3d2d3c7edb
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Cargo.toml
View File

@ -34,7 +34,7 @@ clap = { version = "4.2.4", features = ["derive"] }
cudarc = { version = "0.9.14", features = ["f16"] } cudarc = { version = "0.9.14", features = ["f16"] }
# TODO: Switch back to the official gemm implementation once it has caught up. # TODO: Switch back to the official gemm implementation once it has caught up.
gemm = { version = "0.15.6", package = "candle-gemm" } gemm = { version = "0.15.6", package = "candle-gemm" }
hf-hub = "0.2.0" hf-hub = "0.3.0"
half = { version = "2.3.1", features = ["num-traits", "use-intrinsics", "rand_distr"] } half = { version = "2.3.1", features = ["num-traits", "use-intrinsics", "rand_distr"] }
image = { version = "0.24.7", default-features = false, features = ["jpeg", "png"] } image = { version = "0.24.7", default-features = false, features = ["jpeg", "png"] }
imageproc = { version = "0.23.0", default-features = false } imageproc = { version = "0.23.0", default-features = false }

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 {