optimum/habana/transformers/models/starcoder2/__init__.py (
6
lines of code) (
raw
):
from .modeling_starcoder2 import ( GaudiStarcoder2Attention, GaudiStarcoder2DecoderLayer, GaudiStarcoder2ForCausalLM, GaudiStarcoder2Model, )