From fa255cbc5621538f25abe45a6d372b6d395dba7e Mon Sep 17 00:00:00 2001 From: Tim Dettmers Date: Mon, 13 Feb 2023 17:29:39 -0800 Subject: [PATCH] Added missing import. --- bitsandbytes/nn/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bitsandbytes/nn/__init__.py b/bitsandbytes/nn/__init__.py index 221b5f7..7c2b552 100644 --- a/bitsandbytes/nn/__init__.py +++ b/bitsandbytes/nn/__init__.py @@ -2,4 +2,4 @@ # # This source code is licensed under the MIT license found in the # LICENSE file in the root directory of this source tree. -from .modules import Int8Params, Linear8bitLt, StableEmbedding, OutlierAwareLinear, Fake4bitLinear +from .modules import Int8Params, Linear8bitLt, StableEmbedding, OutlierAwareLinear, Fake4bitLinear, LinearFP8