diff options
| author | Nicolas Patry <patry.nicolas@protonmail.com> | 2023-08-02 19:18:43 +0200 |
|---|---|---|
| committer | Nicolas Patry <patry.nicolas@protonmail.com> | 2023-08-02 19:18:43 +0200 |
| commit | dba31473d40c88fed22574ba96021dc59f25f3f7 (patch) | |
| tree | 5ce0c87397bd15678f7d5df4a6a9c4ab581fde99 /candle-book | |
| parent | 1b2b32e58d13ac96cee42562b845fcecfd3a08de (diff) | |
| download | candle-dba31473d40c88fed22574ba96021dc59f25f3f7.tar.gz candle-dba31473d40c88fed22574ba96021dc59f25f3f7.tar.bz2 candle-dba31473d40c88fed22574ba96021dc59f25f3f7.zip | |
Typos and format and CD only when PR lands.
Diffstat (limited to 'candle-book')
| -rw-r--r-- | candle-book/src/inference/hub.md | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-book/src/inference/hub.md b/candle-book/src/inference/hub.md index a974a1fa..b924b76d 100644 --- a/candle-book/src/inference/hub.md +++ b/candle-book/src/inference/hub.md @@ -67,8 +67,8 @@ let bias = weights.get("bert.encoder.layer.0.attention.self.query.bias").unwrap( let linear = Linear::new(weight.clone(), Some(bias.clone())); -let input_ids = Tensor::zeros((3, 7680), DType::F32, &Device::Cpu).unwrap(); -let output = linear.forward(&input_ids); +let input_ids = Tensor::zeros((3, 768), DType::F32, &Device::Cpu).unwrap(); +let output = linear.forward(&input_ids).unwrap(); ``` For a full reference, you can check out the full [bert](https://github.com/LaurentMazare/candle/tree/main/candle-examples/examples/bert) example. |