Open anton-l opened 1 year ago
This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.
Please note that issues that do not follow the contributing guidelines are likely to be ignored.
Let's maybe solve this PR after: https://github.com/huggingface/diffusers/pull/1924
This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.
Please note that issues that do not follow the contributing guidelines are likely to be ignored.
Actually ok for me to only add the PipelineTesterMixin to the following two:
cc @williamberman do you maybe have some time here?
@patrickvonplaten yep!
Closing this as we won't have time in the near future I'm afraid
Context
The #1526 PR implemented a
PipelineTesterMixin
with common fast tests for pipelines to inherit: saving/loading, casting to fp16, checking tuple outputs, attention slicing, etc. To supportPipelineTesterMixin
, a fast tests class has to:pipeline_class
https://github.com/huggingface/diffusers/blob/02d83c9ff1b93f2c6f9c94f9369b3e4bc1ba8ce7/tests/pipelines/stable_diffusion/test_stable_diffusion.py#L46get_dummy_components()
https://github.com/huggingface/diffusers/blob/02d83c9ff1b93f2c6f9c94f9369b3e4bc1ba8ce7/tests/pipelines/stable_diffusion/test_stable_diffusion.py#L48-L53get_dummy_inputs(device, seed)
https://github.com/huggingface/diffusers/blob/02d83c9ff1b93f2c6f9c94f9369b3e4bc1ba8ce7/tests/pipelines/stable_diffusion/test_stable_diffusion.py#L102-L109And that's it! Then the rest of the fast tests can either be trimmed (like checking tuple outputs and fp16, bacause they're already in the
PipelineTesterMixin
), or updated to useget_dummy_components()
andget_dummy_inputs()
instead ofself.dummy_unet
,self.dummy_text_encoder
,self.dummy_image
.The list of tests to update
AltDiffusionPipelineFastTests
(:red_circle: #1629)AltDiffusionImg2ImgPipelineFastTests
AudioDiffusionPipelineFastTests
(currently namedPipelineFastTests
)DanceDiffusionPipelineFastTests
(:yellow_circle: #1630)DDIMPipelineFastTests
DDPMPipelineFastTests
KarrasVePipelineFastTests
LDMTextToImagePipelineFastTests
LDMSuperResolutionPipelineFastTests
LDMPipelineFastTests
PaintByExamplePipelineFastTests
PNDMPipelineFastTests
RepaintPipelineIntegrationTests
ScoreSdeVeipelineFastTests
CycleDiffusionPipelineFastTests
(:yellow_circle: #1630)StableDiffusionPipelineFastTests
StableDiffusionImageVariationPipelineFastTests
StableDiffusionImg2ImgPipelineFastTests
(:yellow_circle: #1630)StableDiffusionInpaintPipelineFastTests
StableDiffusionInpaintLegacyPipelineFastTests
StableDiffusion2PipelineFastTests
StableDiffusiondepth2imgPipelineFastTests
(:red_circle: https://github.com/huggingface/diffusers/issues/1631 https://github.com/huggingface/diffusers/issues/1634)StableDiffusion2InpaintPipelineFastTests
StableDiffusionUpscalePipelineFastTests
StableDiffusion2VPredictionPipelineFastTests
SafeDiffusionPipelineFastTests
VersatileDiffusionDualGuidedPipelineFastTests
VersatileDiffusionImageVariationPipelineFastTests
VersatileDiffusionMegaPipelineFastTests
VersatileDiffusionTextToImagePipelineFastTests
VQDiffusionPipelineFastTests
cc @patrickvonplaten @patil-suraj @pcuenca @williamberman