__init__.py 485 B

123456789
  1. from transformers import AutoConfig, AutoImageProcessor, AutoModelForCausalLM
  2. from .configuration_mineru2 import Mineru2QwenConfig
  3. from .image_processing_mineru2 import Mineru2ImageProcessor
  4. from .modeling_mineru2 import Mineru2QwenForCausalLM
  5. AutoConfig.register(Mineru2QwenConfig.model_type, Mineru2QwenConfig)
  6. AutoModelForCausalLM.register(Mineru2QwenConfig, Mineru2QwenForCausalLM)
  7. AutoImageProcessor.register(Mineru2QwenConfig, slow_image_processor_class=Mineru2ImageProcessor)