From 4f09c80cd83cf75263144a63239155efbeea62d8 Mon Sep 17 00:00:00 2001 From: drbh Date: Thu, 21 Mar 2024 12:34:25 -0400 Subject: [PATCH] fix: prefer spaces url over temp url (#1662) This PR fixes the broken urls in the idefics tests causing CI to fail --- integration-tests/models/test_idefics.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/integration-tests/models/test_idefics.py b/integration-tests/models/test_idefics.py index 7e1d3e11..e49e9c12 100644 --- a/integration-tests/models/test_idefics.py +++ b/integration-tests/models/test_idefics.py @@ -18,7 +18,7 @@ async def idefics(idefics_handle): @pytest.mark.asyncio async def test_idefics(idefics, response_snapshot): response = await idefics.generate( - "User:![](https://temp-5681.s3.us-west-2.amazonaws.com/chicken_on_money.png)Can you tell me a very short story based on the image?", + "User:![](https://huggingface.co/spaces/HuggingFaceM4/idefics_playground/resolve/6af279110440b7ba0c43f6a0be93b9926f062c42/example_images/chicken_on_money.png)Can you tell me a very short story based on the image?", max_new_tokens=10, decoder_input_details=True, ) @@ -31,7 +31,7 @@ async def test_idefics(idefics, response_snapshot): async def test_idefics_load(idefics, generate_load, response_snapshot): responses = await generate_load( idefics, - "User:![](https://temp-5681.s3.us-west-2.amazonaws.com/chicken_on_money.png)Can you tell me a very short story based on the image?", + "User:![](https://huggingface.co/spaces/HuggingFaceM4/idefics_playground/resolve/6af279110440b7ba0c43f6a0be93b9926f062c42/example_images/chicken_on_money.png)Can you tell me a very short story based on the image?", max_new_tokens=10, n=4, )