Skip to content

GitLab

Unverified Commit 1f5f17c5 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

[Versatile Diffusion] Fix tests (#2336)

parent 98c1a8e7
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
tests/pipelines/versatile_diffusion/test_versatile_diffusion_dual_guided.py
View file @ 1f5f17c5
...@@ -48,7 +48,7 @@ class VersatileDiffusionDualGuidedPipelineIntegrationTests(unittest.TestCase): ...@@ -48,7 +48,7 @@ class VersatileDiffusionDualGuidedPipelineIntegrationTests(unittest.TestCase):
pipe.set_progress_bar_config(disable=None) pipe.set_progress_bar_config(disable=None)
second_prompt = load_image( 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) generator = torch.manual_seed(0)
...@@ -90,7 +90,7 @@ class VersatileDiffusionDualGuidedPipelineIntegrationTests(unittest.TestCase): ...@@ -90,7 +90,7 @@ class VersatileDiffusionDualGuidedPipelineIntegrationTests(unittest.TestCase):
first_prompt = "cyberpunk 2077" first_prompt = "cyberpunk 2077"
second_prompt = load_image( 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) generator = torch.manual_seed(0)
image = pipe( image = pipe(
......
tests/pipelines/versatile_diffusion/test_versatile_diffusion_image_variation.py
View file @ 1f5f17c5
...@@ -38,7 +38,7 @@ class VersatileDiffusionImageVariationPipelineIntegrationTests(unittest.TestCase ...@@ -38,7 +38,7 @@ class VersatileDiffusionImageVariationPipelineIntegrationTests(unittest.TestCase
pipe.set_progress_bar_config(disable=None) pipe.set_progress_bar_config(disable=None)
image_prompt = load_image( 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) generator = torch.manual_seed(0)
image = pipe( image = pipe(
......
tests/pipelines/versatile_diffusion/test_versatile_diffusion_mega.py
View file @ 1f5f17c5
...@@ -46,7 +46,7 @@ class VersatileDiffusionMegaPipelineIntegrationTests(unittest.TestCase): ...@@ -46,7 +46,7 @@ class VersatileDiffusionMegaPipelineIntegrationTests(unittest.TestCase):
pipe.set_progress_bar_config(disable=None) pipe.set_progress_bar_config(disable=None)
prompt_image = load_image( 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) generator = torch.manual_seed(0)
...@@ -86,7 +86,7 @@ class VersatileDiffusionMegaPipelineIntegrationTests(unittest.TestCase): ...@@ -86,7 +86,7 @@ class VersatileDiffusionMegaPipelineIntegrationTests(unittest.TestCase):
prompt = "cyberpunk 2077" prompt = "cyberpunk 2077"
init_image = load_image( 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) generator = torch.manual_seed(0)
image = pipe.dual_guided( image = pipe.dual_guided(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment