| 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"] |