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

Fix PIL dep for tests (#34028)

Browse Source 
Fix PIL dep for tess
This commit is contained in:
Zach Mueller
2024-10-09 10:45:06 -04:00
committed by GitHub
parent 5ee52ae0bc
commit 4fb28703ad
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/trainer/test_trainer.py
View File

@@ -92,6 +92,7 @@ from transformers.testing_utils import (
require_torch_tf32, require_torch_tf32,
require_torch_up_to_2_accelerators, require_torch_up_to_2_accelerators,
require_torchdynamo, require_torchdynamo,
require_vision,
require_wandb, require_wandb,
slow, slow,
torch_device, torch_device,
@@ -3812,6 +3813,7 @@ class TrainerIntegrationTest(TestCasePlus, TrainerIntegrationCommon):
reloaded_tokenizer(test_sentence, padding="max_length").input_ids, reloaded_tokenizer(test_sentence, padding="max_length").input_ids,
) )
@require_vision
def test_trainer_saves_image_processor(self): def test_trainer_saves_image_processor(self):
MODEL_ID = "openai/clip-vit-base-patch32" MODEL_ID = "openai/clip-vit-base-patch32"
image_processor = AutoImageProcessor.from_pretrained(MODEL_ID) image_processor = AutoImageProcessor.from_pretrained(MODEL_ID)
@@ -3845,6 +3847,7 @@ class TrainerIntegrationTest(TestCasePlus, TrainerIntegrationCommon):
self.assertDictEqual(feature_extractor.to_dict(), reloaded_feature_extractor.to_dict()) self.assertDictEqual(feature_extractor.to_dict(), reloaded_feature_extractor.to_dict())
@require_vision
def test_trainer_saves_processor(self): def test_trainer_saves_processor(self):
MODEL_ID = "openai/clip-vit-base-patch32" MODEL_ID = "openai/clip-vit-base-patch32"
image_processor = AutoImageProcessor.from_pretrained(MODEL_ID) image_processor = AutoImageProcessor.from_pretrained(MODEL_ID)