diff options
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}") |