SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Add ViTImageProcessorFast to tests (#31424)

Browse Source 
* Add ViTImageProcessor to tests

* Correct data format

* Review comments
This commit is contained in:
amyeroberts
2024-06-25 13:36:58 +01:00
committed by GitHub
parent aab0829790
commit 0f67ba1d74
26 changed files with 230 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/models/video_llava/test_image_processing_video_llava.py
View File

@@ -285,7 +285,7 @@ class VideoLlavaImageProcessingTest(ImageProcessingTestMixin, unittest.TestCase)
encoded_images = image_processor(
image_inputs[0],
return_tensors="pt",
input_data_format="channels_first",
input_data_format="channels_last",
image_mean=0,
image_std=1,
).pixel_values_images
@@ -296,7 +296,7 @@ class VideoLlavaImageProcessingTest(ImageProcessingTestMixin, unittest.TestCase)
encoded_images = image_processor(
image_inputs,
return_tensors="pt",
input_data_format="channels_first",
input_data_format="channels_last",
image_mean=0,
image_std=1,
).pixel_values_images