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, )