Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Tests] Correct image folder tests (#427) 

* [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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 6 deletions
Show Stats 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"