aphrodite-engine/aphrodite/modeling/models/__init__.py

from aphrodite.modeling.models.llama import LlamaForCausalLM
from aphrodite.modeling.models.mistral import MistralForCausalLM
from aphrodite.modeling.models.gpt_j import GPTJForCausalLM
from aphrodite.modeling.models.gpt_neox import GPTNeoXForCausalLM

__all__ = [
    "LlamaForCausalLM",
    "GPTJForCausalLM",
    "GPTNeoXForCausalLM",
    "MistralForCausalLM",
]