__init__.py 761 B

123456789101112131415161718192021
  1. from sglang.srt.configs.model_config import multimodal_model_archs
  2. from sglang.srt.models.registry import ModelRegistry
  3. try:
  4. # sglang==0.4.5.post3
  5. from sglang.srt.managers.multimodal_processor import (
  6. PROCESSOR_MAPPING as PROCESSOR_MAPPING,
  7. )
  8. except ImportError:
  9. # sglang==0.4.4.post1
  10. from sglang.srt.managers.image_processor import (
  11. IMAGE_PROCESSOR_MAPPING as PROCESSOR_MAPPING,
  12. )
  13. from .. import vlm_hf_model as _
  14. from .image_processor import Mineru2ImageProcessor
  15. from .model import Mineru2QwenForCausalLM
  16. ModelRegistry.models[Mineru2QwenForCausalLM.__name__] = Mineru2QwenForCausalLM
  17. PROCESSOR_MAPPING[Mineru2QwenForCausalLM] = Mineru2ImageProcessor
  18. multimodal_model_archs.append(Mineru2QwenForCausalLM.__name__)