File size: 967 Bytes
1d43c12 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
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"]
|