From be5164807233eebae77d41c3f84b7d5b1455b453 Mon Sep 17 00:00:00 2001 From: aria-hacker Date: Fri, 4 Oct 2024 07:32:35 +0800 Subject: [PATCH] Add warning for unsupported 'use_fast' parameter in AriaProcessor --- aria/model/processing_aria.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/aria/model/processing_aria.py b/aria/model/processing_aria.py index afbcc4b..c425c5d 100644 --- a/aria/model/processing_aria.py +++ b/aria/model/processing_aria.py @@ -18,6 +18,7 @@ # under the License. import inspect +import logging import re from typing import List, Optional, Union @@ -34,6 +35,8 @@ from .vision_processor import AriaVisionProcessor +logger = logging.getLogger(__name__) + class AriaProcessor(ProcessorMixin): """ @@ -229,6 +232,7 @@ def from_pretrained( **cls._extract_kwargs(AriaVisionProcessor.from_pretrained, **kwargs), ) if "use_fast" in kwargs: + logger.warning("use_fast is not supported for AriaProcessor. Ignoring...") kwargs.pop("use_fast") try: tokenizer = AutoTokenizer.from_pretrained(