From 57cd6f63bfccb6b184ab36a296d027ba80148859 Mon Sep 17 00:00:00 2001 From: Joshua Rosenkranz Date: Wed, 6 Aug 2025 19:44:30 +0000 Subject: [PATCH] fixed fp8 import so as to not require user to import Signed-off-by: Joshua Rosenkranz --- fms_mo/aiu_addons/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/fms_mo/aiu_addons/__init__.py b/fms_mo/aiu_addons/__init__.py index 1d9c5cf..3ab3f3b 100644 --- a/fms_mo/aiu_addons/__init__.py +++ b/fms_mo/aiu_addons/__init__.py @@ -30,6 +30,7 @@ def _infer_quantization_config(quant_config: dict) -> dict | None: ) # First, import required FP8 linear classes from fms-mo # Local + import fms_mo.aiu_addons.fp8.fp8_attn # pylint: disable=unused-import import fms_mo.aiu_addons.fp8.fp8_adapter # pylint: disable=unused-import import fms_mo.aiu_addons.fp8.fp8_linear # pylint: disable=unused-import