typo

mosaicml · Dec 29, 2023 · 5c4f3e7 · 5c4f3e7
1 parent 70de92f
commit 5c4f3e7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)