recast_llama/__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 (
        RECAST_llamaModel,
        RECAST_LlamaForCausalLM,
    )
    from .configuration_recast_llama import RECAST_llama

from transformers import AutoConfig, AutoModel, AutoModelForCausalLM

# Register your models with Auto classes
AutoConfig.register("recast_llama", RECAST_llama)
AutoModel.register(RECAST_llama, RECAST_llamaModel)
AutoModelForCausalLM.register(RECAST_llama, RECAST_LlamaForCausalLM)

_import_structure = {
    "configuration_recastmlp_llama": ["RECAST_llama"],
    "modeling_recastmlp_llama": ["RECAST_llamaModel", "RECAST_LlamaForCausalLM"],
}

__all__ = ["RECAST_llamaModel", "RECAST_LlamaForCausalLM", "RECAST_llama"]