Skip to content

Commit

Permalink
Fix the book.
Browse files Browse the repository at this point in the history
  • Loading branch information
LaurentMazare committed Oct 26, 2024
1 parent c7835e1 commit 7579a1b
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions candle-book/src/inference/hub.md
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,8 @@ Then let's start by downloading the [model file](https://huggingface.co/bert-bas

```rust
# extern crate candle_core;
# extern crate hf_hub;
use hf_hub::api::sync::Api;
# extern crate candle_hf_hub;
use candle_hf_hub::api::sync::Api;
use candle_core::Device;

let api = Api::new().unwrap();
Expand Down Expand Up @@ -50,8 +50,8 @@ Now that we have our weights, we can use them in our bert architecture:
```rust
# extern crate candle_core;
# extern crate candle_nn;
# extern crate hf_hub;
# use hf_hub::api::sync::Api;
# extern crate candle_hf_hub;
# use candle_hf_hub::api::sync::Api;
#
# let api = Api::new().unwrap();
# let repo = api.model("bert-base-uncased".to_string());
Expand Down

0 comments on commit 7579a1b

Please sign in to comment.