AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

[Tests] Correct image folder tests (#427)

Browse Source 
* [Tests] Correct image folder tests

* up
This commit is contained in:
Patrick von Platen
2022-09-08 16:45:29 +02:00
committed by GitHub
parent f8325cfd7b
commit f55190b275
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_pipelines.py
View File

@@ -1158,11 +1158,10 @@ class PipelineTesterMixin(unittest.TestCase):
model_id = "CompVis/stable-diffusion-v1-4"
pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
model_id,
revision="fp16", # fp16 to infer 768x512 images with 16GB of VRAM
torch_dtype=torch.float16,
use_auth_token=True,
)
pipe.to(torch_device)
pipe.enable_attention_slicing()
pipe.set_progress_bar_config(disable=None)
prompt = "A fantasy landscape, trending on artstation"
@@ -1204,10 +1203,9 @@ class PipelineTesterMixin(unittest.TestCase):
pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
model_id,
scheduler=lms,
revision="fp16", # fp16 to infer 768x512 images with 16GB of VRAM
torch_dtype=torch.float16,
use_auth_token=True,
)
pipe.enable_attention_slicing()
pipe.to(torch_device)
pipe.set_progress_bar_config(disable=None)
@@ -1252,11 +1250,10 @@ class PipelineTesterMixin(unittest.TestCase):
model_id = "CompVis/stable-diffusion-v1-4"
pipe = StableDiffusionInpaintPipeline.from_pretrained(
model_id,
revision="fp16", # fp16 to infer 768x512 images in 16GB of VRAM
torch_dtype=torch.float16,
use_auth_token=True,
)
pipe.to(torch_device)
pipe.enable_attention_slicing()
pipe.set_progress_bar_config(disable=None)
prompt = "A red cat sitting on a parking bench"