Skip to content

Commit 1f5f17c

Browse files
[Versatile Diffusion] Fix tests (#2336)
1 parent 98c1a8e commit 1f5f17c

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

tests/pipelines/versatile_diffusion/test_versatile_diffusion_dual_guided.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def test_remove_unused_weights_save_load(self):
4848
pipe.set_progress_bar_config(disable=None)
4949

5050
second_prompt = load_image(
51-
"https://raw.githubusercontent.com/SHI-Labs/Versatile-Diffusion/master/assets/benz.jpg"
51+
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/versatile_diffusion/benz.jpg"
5252
)
5353

5454
generator = torch.manual_seed(0)
@@ -90,7 +90,7 @@ def test_inference_dual_guided(self):
9090

9191
first_prompt = "cyberpunk 2077"
9292
second_prompt = load_image(
93-
"https://raw.githubusercontent.com/SHI-Labs/Versatile-Diffusion/master/assets/benz.jpg"
93+
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/versatile_diffusion/benz.jpg"
9494
)
9595
generator = torch.manual_seed(0)
9696
image = pipe(

tests/pipelines/versatile_diffusion/test_versatile_diffusion_image_variation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def test_inference_image_variations(self):
3838
pipe.set_progress_bar_config(disable=None)
3939

4040
image_prompt = load_image(
41-
"https://raw.githubusercontent.com/SHI-Labs/Versatile-Diffusion/master/assets/benz.jpg"
41+
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/versatile_diffusion/benz.jpg"
4242
)
4343
generator = torch.manual_seed(0)
4444
image = pipe(

tests/pipelines/versatile_diffusion/test_versatile_diffusion_mega.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def test_from_save_pretrained(self):
4646
pipe.set_progress_bar_config(disable=None)
4747

4848
prompt_image = load_image(
49-
"https://raw.githubusercontent.com/SHI-Labs/Versatile-Diffusion/master/assets/benz.jpg"
49+
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/blob/main/versatile_diffusion/benz.jpg"
5050
)
5151

5252
generator = torch.manual_seed(0)
@@ -86,7 +86,7 @@ def test_inference_dual_guided_then_text_to_image(self):
8686

8787
prompt = "cyberpunk 2077"
8888
init_image = load_image(
89-
"https://raw.githubusercontent.com/SHI-Labs/Versatile-Diffusion/master/assets/benz.jpg"
89+
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/blob/main/versatile_diffusion/benz.jpg"
9090
)
9191
generator = torch.manual_seed(0)
9292
image = pipe.dual_guided(

0 commit comments

Comments
 (0)