Skip to content

Commit

Permalink
Merge pull request #12 from rhymes-ai/log
Browse files Browse the repository at this point in the history
Add warning for unsupported 'use_fast' parameter in AriaProcessor
  • Loading branch information
aria-hacker authored Oct 3, 2024
2 parents 2f6420a + be51648 commit 99b879c
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 99b879c

Please sign in to comment.