remove importing fp8 modules in fp8's __init__ because it causes cicd…

… to fail
huggingface · Jan 17, 2025 · 7c7949c · 7c7949c
1 parent f70230e
commit 7c7949c
Showing 1 changed file with 0 additions and 11 deletions.
diff --git a/src/nanotron/fp8/__init__.py b/src/nanotron/fp8/__init__.py
@@ -1,11 +0,0 @@
-import warnings
-
-from nanotron.fp8.dtypes import DTypes  # noqa
-from nanotron.fp8.linear import FP8Linear  # noqa
-from nanotron.fp8.tensor import FP8Tensor  # noqa
-
-try:
-    import transformer_engine as te  # noqa
-    import transformer_engine_torch as tex  # noqa
-except ImportError:
-    warnings.warn("Please install Transformer engine for FP8 training!")