2022-08-01 10:31:48 +00:00
|
|
|
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
|
|
|
#
|
|
|
|
|
# This source code is licensed under the MIT license found in the
|
2021-10-06 02:16:20 +00:00
|
|
|
# LICENSE file in the root directory of this source tree.
|
2023-05-07 20:34:03 +00:00
|
|
|
from .modules import Int8Params, Linear8bitLt, StableEmbedding, Linear4bit, LinearNF4, LinearFP4, Params4bit, OutlierAwareLinear, SwitchBackLinearBnb
|
2023-04-12 20:41:30 +00:00
|
|
|
from .triton_based_modules import SwitchBackLinear, SwitchBackLinearGlobal, SwitchBackLinearVectorwise, StandardLinear
|
