From 249b652939e310e73edd2d960f2c130386f85244 Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Mon, 25 Nov 2024 15:47:51 +0100 Subject: [PATCH] fix --- tests/configs/_image_text_to_text_.yaml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/configs/_image_text_to_text_.yaml b/tests/configs/_image_text_to_text_.yaml index 20043a67..4404d6e3 100644 --- a/tests/configs/_image_text_to_text_.yaml +++ b/tests/configs/_image_text_to_text_.yaml @@ -3,8 +3,8 @@ hydra: sweeper: params: backend.task: image-text-to-text - backend.model: hf-internal-testing/tiny-random-GitForCausalLM, + backend.model: hf-internal-testing/tiny-random-BlipForConditionalGeneration, - hf-internal-testing/tiny-random-Blip2ForConditionalGeneration, - hf-internal-testing/tiny-random-IdeficsForVisionText2Text + hf-internal-testing/tiny-random-IdeficsForVisionText2Text, + hf-internal-testing/tiny-random-GitForCausalLM +scenario.input_shapes.num_images: 2