optimum/habana/transformers/models/mamba/__init__.py (
4
lines of code) (
raw
):
from .modeling_mamba import ( gaudi_MambaForCausalLM_prepare_inputs_for_generation, gaudi_MambaForCausalLM_update_model_kwargs_for_generation, )