[Model] Support Pixtral models in the HF Transformers format (#9036)

This commit is contained in:
Michael Goin
2024-10-18 15:29:56 -04:00
committed by GitHub
parent 67a7e5ef38
commit 3921a2f29e
7 changed files with 503 additions and 12 deletions

View File

@@ -1,3 +1,4 @@
from functools import lru_cache
from typing import Any, cast
@@ -37,6 +38,9 @@ def get_processor(
return cast(ProcessorMixin, processor)
cached_get_processor = lru_cache(get_processor)
def get_image_processor(
processor_name: str,
*args: Any,