File size: 960 Bytes
bcc6605 | 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 33 34 | from transformers import AutoConfig, AutoModel, AutoModelForCausalLM
from .configuration_vectorllm import VectorLLMConfig
from .hf_model import RADIOConfig, RADIOModel
from .image_processing_vectorllm import VectorLLMImageProcessor
from .modeling_vectorllm import VectorLLMForCausalLM
from .processing_vectorllm import VectorLLMProcessor
def _safe_register(register_fn, *args):
try:
register_fn(*args)
except ValueError:
pass
def bootstrap_local_registry():
_safe_register(AutoConfig.register, VectorLLMConfig.model_type, VectorLLMConfig)
_safe_register(AutoModel.register, VectorLLMConfig, VectorLLMForCausalLM)
_safe_register(AutoModelForCausalLM.register, VectorLLMConfig, VectorLLMForCausalLM)
bootstrap_local_registry()
__all__ = [
"RADIOConfig",
"RADIOModel",
"VectorLLMConfig",
"VectorLLMForCausalLM",
"VectorLLMImageProcessor",
"VectorLLMProcessor",
"bootstrap_local_registry",
]
|