tests/test_pipelines_common.py · 2b4f849db9cfd73c5c367b2ac124c8e48ef32430 · renzhc / diffusers_dcu

[PipelineTesterMixin] Handle non-image outputs for attn slicing test (#2504) · 2b4f849d

Sanchit Gandhi authored Mar 06, 2023



* [PipelineTesterMixin] Handle non-image outputs for batch/sinle inference test

* style

---------
Co-authored-by: William Berman <WLBberman@gmail.com>

2b4f849d

test_pipelines_common.py 23.3 KB

Replace test_pipelines_common.py