| from transformers.models.auto import ( |
| AutoConfig, |
| AutoModel, |
| AutoProcessor, |
| AutoImageProcessor, |
| ) |
|
|
| from .configuration_ax_ve import AXVEConfig |
| from .configuration_ax_ve import AXVEVisionConfig |
|
|
| from .modeling_ax_ve import AXVEVisionModel |
| from .processing_ax_ve import AXVEProcessor |
| from .image_processing_ax_ve import AXVEImageProcessor |
|
|
| AutoConfig.register(AXVEConfig.model_type, AXVEConfig) |
| AutoConfig.register(AXVEVisionConfig.model_type, AXVEVisionConfig) |
| AutoModel.register(AXVEVisionConfig, AXVEVisionModel) |
| AutoProcessor.register(AXVEConfig, AXVEProcessor) |
| AutoImageProcessor.register(AXVEConfig, AXVEImageProcessor) |
|
|
| print("A.X VE vision encoder registered") |
|
|
