| from transformers import ( | |
| AutoModelForCausalLM, | |
| AutoProcessor, | |
| AutoImageProcessor, | |
| AutoConfig, | |
| ) | |
| from .processor import ( | |
| Qwen2VLImagePointerProcessor, | |
| get_processor, | |
| V1Processor, | |
| collate_fn, | |
| ) | |
| from .modeling_v1 import V1ForConditionalGeneration | |
| from .configuration_v1 import V1Config | |
| print("Registering V1 model and processor with Transformers") | |
| AutoConfig.register("v1", V1Config) | |
| AutoModelForCausalLM.register( | |
| V1Config, V1ForConditionalGeneration | |
| ) | |
| AutoProcessor.register(V1Config, V1Processor) | |
| AutoImageProcessor.register(V1Config, Qwen2VLImagePointerProcessor) | |