From 07af87a1d801852645966d89bd193808ff7c5b35 Mon Sep 17 00:00:00 2001 From: Laurent Mazare Date: Fri, 13 Oct 2023 17:21:20 +0200 Subject: Typos. (#1084) --- candle-pyo3/e5.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'candle-pyo3') 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}") -- cgit v1.2.3