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 (
        RECAST8b_llamaModel,
        RECAST8b_LlamaForCausalLM,
    )
    from .configuration_recast_llama import RECAST8b_llama

from transformers import AutoConfig, AutoModel, AutoModelForCausalLM

# Register your models with Auto classes
AutoConfig.register("recast8b_llama", RECAST8b_llama)
AutoModel.register(RECAST8b_llama, RECAST8b_llamaModel)
AutoModelForCausalLM.register(RECAST8b_llama, RECAST8b_LlamaForCausalLM)

_import_structure = {
    "configuration_recastmlp_llama": ["RECAST8b_llama"],
    "modeling_recastmlp_llama": ["RECAST8b_llamaModel", "RECAST8b_LlamaForCausalLM"],
}

__all__ = ["RECAST8b_llamaModel", "RECAST8b_LlamaForCausalLM", "RECAST8b_llama"]