diff options
Diffstat (limited to 'candle-examples/examples/bert/main.rs')
-rw-r--r-- | candle-examples/examples/bert/main.rs | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/candle-examples/examples/bert/main.rs b/candle-examples/examples/bert/main.rs index 6672ad09..79c78968 100644 --- a/candle-examples/examples/bert/main.rs +++ b/candle-examples/examples/bert/main.rs @@ -69,10 +69,11 @@ impl Args { ) } else { let api = Api::new()?; + let api = api.repo(repo); ( - api.get(&repo, "config.json")?, - api.get(&repo, "tokenizer.json")?, - api.get(&repo, "model.safetensors")?, + api.get("config.json")?, + api.get("tokenizer.json")?, + api.get("model.safetensors")?, ) }; let config = std::fs::read_to_string(config_filename)?; |