From f3124a0be0aab1d02ed5aa0ef9d4157f9ed5f986 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 13 Feb 2023 16:53:05 +0000 Subject: [PATCH] [Versatile Diffusion] Fix tests --- .../test_versatile_diffusion_dual_guided.py | 4 ++-- .../test_versatile_diffusion_image_variation.py | 2 +- .../versatile_diffusion/test_versatile_diffusion_mega.py | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/pipelines/versatile_diffusion/test_versatile_diffusion_dual_guided.py b/tests/pipelines/versatile_diffusion/test_versatile_diffusion_dual_guided.py index 82fd037cec73..5edebdde48ba 100644 --- a/tests/pipelines/versatile_diffusion/test_versatile_diffusion_dual_guided.py +++ b/tests/pipelines/versatile_diffusion/test_versatile_diffusion_dual_guided.py @@ -48,7 +48,7 @@ def test_remove_unused_weights_save_load(self): pipe.set_progress_bar_config(disable=None) second_prompt = load_image( - "https://raw.githubusercontent.com/SHI-Labs/Versatile-Diffusion/master/assets/benz.jpg" + "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/versatile_diffusion/benz.jpg" ) generator = torch.manual_seed(0) @@ -90,7 +90,7 @@ def test_inference_dual_guided(self): first_prompt = "cyberpunk 2077" second_prompt = load_image( - "https://raw.githubusercontent.com/SHI-Labs/Versatile-Diffusion/master/assets/benz.jpg" + "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/versatile_diffusion/benz.jpg" ) generator = torch.manual_seed(0) image = pipe( diff --git a/tests/pipelines/versatile_diffusion/test_versatile_diffusion_image_variation.py b/tests/pipelines/versatile_diffusion/test_versatile_diffusion_image_variation.py index a55184429681..19a02b77c9e0 100644 --- a/tests/pipelines/versatile_diffusion/test_versatile_diffusion_image_variation.py +++ b/tests/pipelines/versatile_diffusion/test_versatile_diffusion_image_variation.py @@ -38,7 +38,7 @@ def test_inference_image_variations(self): pipe.set_progress_bar_config(disable=None) image_prompt = load_image( - "https://raw.githubusercontent.com/SHI-Labs/Versatile-Diffusion/master/assets/benz.jpg" + "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/versatile_diffusion/benz.jpg" ) generator = torch.manual_seed(0) image = pipe( diff --git a/tests/pipelines/versatile_diffusion/test_versatile_diffusion_mega.py b/tests/pipelines/versatile_diffusion/test_versatile_diffusion_mega.py index 6597335e2289..8bca6ce5de0d 100644 --- a/tests/pipelines/versatile_diffusion/test_versatile_diffusion_mega.py +++ b/tests/pipelines/versatile_diffusion/test_versatile_diffusion_mega.py @@ -46,7 +46,7 @@ def test_from_save_pretrained(self): pipe.set_progress_bar_config(disable=None) prompt_image = load_image( - "https://raw.githubusercontent.com/SHI-Labs/Versatile-Diffusion/master/assets/benz.jpg" + "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/blob/main/versatile_diffusion/benz.jpg" ) generator = torch.manual_seed(0) @@ -86,7 +86,7 @@ def test_inference_dual_guided_then_text_to_image(self): prompt = "cyberpunk 2077" init_image = load_image( - "https://raw.githubusercontent.com/SHI-Labs/Versatile-Diffusion/master/assets/benz.jpg" + "https://huggingface.co/datasets/hf-internal-testing/diffusers-images/blob/main/versatile_diffusion/benz.jpg" ) generator = torch.manual_seed(0) image = pipe.dual_guided(