optimum/habana/accelerate/utils/__init__.py (
1
lines of code) (
raw
):
from .transformer_engine import FP8ContextWrapper, convert_model, get_fp8_recipe