diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-10-13 17:21:20 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-10-13 16:21:20 +0100 |
| commit | 07af87a1d801852645966d89bd193808ff7c5b35 (patch) | |
| tree | 027989ce41cc4c92467d5de9f03c95c2745f04b2 /candle-pyo3 | |
| parent | eefad2b95f284142e5ef45ecb1a49937a082ef12 (diff) | |
| download | candle-07af87a1d801852645966d89bd193808ff7c5b35.tar.gz candle-07af87a1d801852645966d89bd193808ff7c5b35.tar.bz2 candle-07af87a1d801852645966d89bd193808ff7c5b35.zip | |
Typos. (#1084)
Diffstat (limited to 'candle-pyo3')
| -rw-r--r-- | candle-pyo3/e5.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-pyo3/e5.py b/candle-pyo3/e5.py index a0af0c56..8ca48219 100644 --- a/candle-pyo3/e5.py +++ b/candle-pyo3/e5.py @@ -60,7 +60,7 @@ if __name__ == "__main__": loss = torch.nn.L1Loss() error = loss(hf_pooled, candle_pooled).mean().item() - print(f"Mean error between torch-referenze and candle: {error}") + print(f"Mean error between torch-reference and candle: {error}") # Quantize all attention 'weights' quantized_tensors = {} @@ -101,4 +101,4 @@ if __name__ == "__main__": candle_pooled_2 = average_pool(torch.tensor(encoder_out_2.values()), hf_tokenized["attention_mask"]) error = loss(hf_pooled, candle_pooled_2).mean().item() - print(f"Mean error between torch-referenze and quantized-candle: {error}") + print(f"Mean error between torch-reference and quantized-candle: {error}") |