From 932dc5f90226093dcaac766ee07faa1e7b189ebd Mon Sep 17 00:00:00 2001 From: win10 Date: Wed, 30 Oct 2024 20:49:10 +0530 Subject: [PATCH] update readme --- setup.py | 2 +- triformer/__init__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index fd913d7..f61c5af 100644 --- a/setup.py +++ b/setup.py @@ -3,7 +3,7 @@ setup( name = 'triformer', packages = find_packages(), - version = '2.1.8', + version = '3.0.0', license='MIT', description = 'Transformer components in Triton', long_description='This package implements transformer components in Triton.', diff --git a/triformer/__init__.py b/triformer/__init__.py index 9830eb0..20068b5 100644 --- a/triformer/__init__.py +++ b/triformer/__init__.py @@ -1,3 +1,3 @@ -from .mlp import TritonLinear +#from .mlp import TritonLinear from .layernorm import TritonLayerNorm from .softmax import TritonSoftmax