wehub-resource-sync
e06fe8e8c6
Secret Leaks / trufflehog (push) Failing after 1s
Build documentation / build (push) Failing after 1s
Build documentation / build_other_lang (push) Failing after 0s
CodeQL Security Analysis / CodeQL Analysis (push) Failing after 0s
PR CI / pr-ci (push) Failing after 1s
Slow tests on important models (on Push - A10) / Get all modified files (push) Failing after 1s
Slow tests on important models (on Push - A10) / Model CI (push) Has been skipped
Self-hosted runner (benchmark) / Benchmark (aws-g5-4xlarge-cache) (push) Has been cancelled
New model PR merged notification / Notify new model (push) Has been cancelled
Update Transformers metadata / build_and_package (push) Has been cancelled
20 lines
799 B
Python
20 lines
799 B
Python
import unittest
|
||
|
||
from transformers import Owlv2Processor
|
||
from transformers.testing_utils import require_scipy
|
||
|
||
from ...test_processing_common import ProcessorTesterMixin
|
||
|
||
|
||
@require_scipy
|
||
class Owlv2ProcessorTest(ProcessorTesterMixin, unittest.TestCase):
|
||
processor_class = Owlv2Processor
|
||
# Tiny processor created with make_tiny_processor.py from "google/owlv2-base-patch16-ensemble"
|
||
tiny_model_id = "hf-internal-testing/tiny-processor-owlv2"
|
||
|
||
@classmethod
|
||
def _setup_image_processor(cls):
|
||
image_processor_class = cls._get_component_class_from_processor("image_processor")
|
||
# Default size=960×960 produces ~11 MB pixel_values per image. Use 64×64 for tests.
|
||
return image_processor_class.from_pretrained(cls.tiny_model_id, size={"height": 64, "width": 64})
|