__init__.py

from transformers.utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_torch_available,
)

try:
    if not is_torch_available():
        raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
    pass
else:
    from .modeling_recast_llama import (
        RECAST1B_llamaModel,
        RECAST1B_LlamaForCausalLM,
    )
    from .configuration_recast_llama import RECAST1B_llama

from transformers import AutoConfig, AutoModel, AutoModelForCausalLM

# Register your models with Auto classes
AutoConfig.register("recast1b_llama", RECAST1B_llama)
AutoModel.register(RECAST1B_llama, RECAST1B_llamaModel)
AutoModelForCausalLM.register(RECAST1B_llama, RECAST1B_LlamaForCausalLM)

_import_structure = {
    "configuration_recastmlp_llama": ["RECAST1B_llama"],
    "modeling_recastmlp_llama": ["RECAST1B_llamaModel", "RECAST1B_LlamaForCausalLM"],
}

__all__ = ["RECAST1B_llamaModel", "RECAST1B_LlamaForCausalLM", "RECAST1B_llama"]