optimum/habana/transformers/models/falcon/__init__.py (
8
lines of code) (
raw
):
from .modeling_falcon import ( GaudiFalconAttention, GaudiFalconDecoderLayer, GaudiFalconForCausalLM, GaudiFalconMLP, GaudiFalconModel, gaudi_falcon_linear_forward, )