From 5c4f3e77a80636096a007aa21fe18fe8d80dca3f Mon Sep 17 00:00:00 2001 From: Mihir Patel Date: Fri, 29 Dec 2023 14:25:17 -0700 Subject: [PATCH] typo --- llmfoundry/optim/lion8b.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llmfoundry/optim/lion8b.py b/llmfoundry/optim/lion8b.py index f8a33d5f34..2f6eb88694 100644 --- a/llmfoundry/optim/lion8b.py +++ b/llmfoundry/optim/lion8b.py @@ -236,8 +236,8 @@ def __init__(self, data: Optional[torch.Tensor], try_quantize: bool = True): self._f_decode = None if self._try_quantize: from turbo import dequantize_signed, quantize_signed - self._f_encode = quantize8b - self._f_decode = dequantize8b + self._f_encode = quantize_signed + self._f_decode = dequantize_signed if data is not None: self.set_data(data)