Skip to content

Commit

Permalink
Add warning for unsupported 'use_fast' parameter in AriaProcessor
Browse files Browse the repository at this point in the history
  • Loading branch information
aria-hacker committed Oct 3, 2024
1 parent 2f6420a commit be51648
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions aria/model/processing_aria.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
# under the License.

import inspect
import logging
import re
from typing import List, Optional, Union

Expand All @@ -34,6 +35,8 @@

from .vision_processor import AriaVisionProcessor

logger = logging.getLogger(__name__)


class AriaProcessor(ProcessorMixin):
"""
Expand Down Expand Up @@ -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(
Expand Down

0 comments on commit be51648

Please sign in to comment.