From a4e8235b80301cab24976846732109b50891c921 Mon Sep 17 00:00:00 2001 From: Albert Villanova del Moral <8515462+albertvillanova@users.noreply.github.com> Date: Tue, 28 Oct 2025 08:36:51 +0100 Subject: [PATCH] Replace deprecated AutoModelForVision2Seq with AutoModelForImageTextToText --- tests/test_online_dpo_trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_online_dpo_trainer.py b/tests/test_online_dpo_trainer.py index b5c9a1d9ee1..09f986ad558 100644 --- a/tests/test_online_dpo_trainer.py +++ b/tests/test_online_dpo_trainer.py @@ -38,7 +38,7 @@ if is_vision_available(): import numpy as np from PIL import Image - from transformers import AutoModelForVision2Seq, AutoProcessor + from transformers import AutoModelForImageTextToText, AutoProcessor class TestOnlineDPOTrainer(TrlTestCase): @@ -510,7 +510,7 @@ def test_online_dpo_vlm_trainer(self, model_id): dataset = Dataset.from_dict(dataset_dict) dataset = dataset.cast_column("images", features.Sequence(features.Image())) - model = AutoModelForVision2Seq.from_pretrained(model_id) + model = AutoModelForImageTextToText.from_pretrained(model_id) reward_model = AutoModelForSequenceClassification.from_pretrained( "trl-internal-testing/tiny-LlamaForCausalLM-3.2", num_labels=1 )