bc161d5d06
* delete deprecated stuff * fix copies * remove unused tests * fix modernbert and fuyu * Update src/transformers/cache_utils.py Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com> * bye bye `seen_tokens` * address comments * update typings * ecnoder decoder models follow same pattern as whisper * fix copies * why is it set to False? * fix switch transformers * fix encoder decoder models shared weight * fix copies and RAG * remove `next_cache` * fix gptj/git * fix copies * fix copies * style... * another forgotten docsrting --------- Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>
24 lines
656 B
Python
24 lines
656 B
Python
import shutil
|
|
import tempfile
|
|
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
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.tmpdirname = tempfile.mkdtemp()
|
|
processor = cls.processor_class.from_pretrained("google/owlv2-base-patch16-ensemble")
|
|
processor.save_pretrained(cls.tmpdirname)
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
shutil.rmtree(cls.tmpdirname, ignore_errors=True)
|