AstrAI/astrai/extension/__init__.py

14 lines
302 B
Python

import importlib
import logging
logger = logging.getLogger(__name__)
available: dict[str, bool] = {}
for _name in ["gqa_decode_attn"]:
try:
importlib.import_module(f".{_name}", package=__package__)
available[_name] = True
except ImportError:
available[_name] = False