riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-03 21:00:08 +06:00
Code Issues Actions 25 Packages Projects Releases Wiki Activity

[Tests] Cleanup Janus Testcase (#38311)

Browse Source 
* Cleanup janus testcase

* shift code to setup
This commit is contained in:
Yaswanth Gali 2025-05-23 12:59:16 +05:30 committed by GitHub
parent b59386dc0a
commit 2b585419b4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/models/janus/test_processor_janus.py
View File

@ -20,7 +20,6 @@ import unittest
import numpy as np
from transformers import AutoProcessor, AutoTokenizer, JanusProcessor
from transformers.models.janus.convert_janus_weights_to_hf import CHAT_TEMPLATE
from transformers.utils import is_vision_available
from ...test_processing_common import ProcessorTesterMixin
@ -45,6 +44,9 @@ class JanusProcessorTest(ProcessorTesterMixin, unittest.TestCase):
"deepseek-community/Janus-Pro-1B",
extra_special_tokens=special_image_tokens,
)
# Set the processor to use the default system prompt to False as it's used based on input modality.
# Hence set to False to avoid any issues in the test irrespective of inputs.
processor.use_default_system_prompt = False
processor.save_pretrained(self.tmpdirname)
def get_tokenizer(self, **kwargs):
@ -53,11 +55,8 @@ class JanusProcessorTest(ProcessorTesterMixin, unittest.TestCase):
def get_image_processor(self, **kwargs):
return AutoProcessor.from_pretrained(self.tmpdirname, **kwargs).image_processor
def prepare_processor_dict(self):
# similar to Emu3 and Qwen2VLProcessorTest, but keep the template in the convert script to avoid duplicated code
return {
"chat_template": CHAT_TEMPLATE,
}
def get_processor(self):
return AutoProcessor.from_pretrained(self.tmpdirname)
def test_chat_template_single(self):
"""