diff --git a/.ai/references/testing.md b/.ai/references/testing.md index 62e1ca986a07..999b84804c52 100644 --- a/.ai/references/testing.md +++ b/.ai/references/testing.md @@ -24,7 +24,7 @@ Follow the style introduced in [#14113](https://github.com/huggingface/diffusers - Set `pipeline_class`, `required_input_params_in_call_signature` (params that must appear in `__call__`'s signature), and `batch_input_params` (params that get batched). Use the canonical sets in `..pipeline_params` where one fits, or an inline `frozenset([...])`. - Set `output_shape` — the per-sample output shape for `get_dummy_inputs()`, i.e. `(channels, height, width)` for an image pipeline and `(num_frames, channels, height, width)` for a video one. Assert against `self.output_shape` in pipeline-specific tests instead of repeating the literal. - Implement `get_dummy_components(...)` — build every sub-module from the **real classes** at tiny config, each preceded by `torch.manual_seed(0)`. - - Implement `get_dummy_inputs()` — **no `device` / `seed` arguments** (unlike the old style). Use `self.get_generator(0)` for the generator, keep sizes tiny, and set `output_type="pt"` so tests compare torch tensors directly with `assert_tensors_close` (no numpy round-trip). Remember `"pt"` images are `(batch, channels, height, width)`. + - Implement `get_dummy_inputs()` — **no `device` / `seed` arguments** (unlike the old style). Use `self.get_generator(0)` for the generator, keep sizes tiny, and set `output_type="pt"` so tests compare torch tensors directly with `assert_tensors_close` (no numpy round-trip). Remember `"pt"` images are `(batch, channels, height, width)` and videos `(batch, num_frames, channels, height, width)`, unlike `"np"` outputs which are `(batch, height, width, channels)`. - **Compose the config with one mixin per concern**, one test class each, named `Test...`. Add only the mixins that apply: - `PipelineTesterMixin` — core save/load, dict-vs-tuple equivalence, batching, dtype/device, callbacks. Put pipeline-specific tests as methods on this class. - `MemoryTesterMixin` — CPU offload, group offload, layerwise casting. diff --git a/tests/pipelines/ace_step/test_ace_step.py b/tests/pipelines/ace_step/test_ace_step.py index 5ced68691272..d92f5ed167ee 100644 --- a/tests/pipelines/ace_step/test_ace_step.py +++ b/tests/pipelines/ace_step/test_ace_step.py @@ -237,7 +237,6 @@ def get_dummy_inputs(self): "num_inference_steps": 2, "generator": self.get_generator(0), "max_text_length": 32, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/allegro/test_allegro.py b/tests/pipelines/allegro/test_allegro.py index 8bc1bfb08faf..9f74c98a85e6 100644 --- a/tests/pipelines/allegro/test_allegro.py +++ b/tests/pipelines/allegro/test_allegro.py @@ -129,7 +129,6 @@ def get_dummy_inputs(self): "width": 16, "num_frames": 8, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/animatediff/test_animatediff.py b/tests/pipelines/animatediff/test_animatediff.py index 45c13bbd2c11..5162710be342 100644 --- a/tests/pipelines/animatediff/test_animatediff.py +++ b/tests/pipelines/animatediff/test_animatediff.py @@ -117,7 +117,6 @@ def get_dummy_inputs(self): "generator": self.get_generator(0), "num_inference_steps": 2, "guidance_scale": 7.5, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/animatediff/test_animatediff_controlnet.py b/tests/pipelines/animatediff/test_animatediff_controlnet.py index 5946a86ace41..d7be7b15bac6 100644 --- a/tests/pipelines/animatediff/test_animatediff_controlnet.py +++ b/tests/pipelines/animatediff/test_animatediff_controlnet.py @@ -126,7 +126,6 @@ def get_dummy_inputs(self, num_frames: int = 2): "num_inference_steps": 2, "num_frames": num_frames, "guidance_scale": 7.5, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/animatediff/test_animatediff_sdxl.py b/tests/pipelines/animatediff/test_animatediff_sdxl.py index 8a42c9dd3653..6fc3cdb93b7f 100644 --- a/tests/pipelines/animatediff/test_animatediff_sdxl.py +++ b/tests/pipelines/animatediff/test_animatediff_sdxl.py @@ -112,7 +112,6 @@ def get_dummy_inputs(self): "generator": self.get_generator(0), "num_inference_steps": 2, "guidance_scale": 7.5, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/animatediff/test_animatediff_sparsectrl.py b/tests/pipelines/animatediff/test_animatediff_sparsectrl.py index 967183802bec..220f3f5625be 100644 --- a/tests/pipelines/animatediff/test_animatediff_sparsectrl.py +++ b/tests/pipelines/animatediff/test_animatediff_sparsectrl.py @@ -128,7 +128,6 @@ def get_dummy_inputs(self, num_frames: int = 2): "num_inference_steps": 2, "num_frames": num_frames, "guidance_scale": 7.5, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/animatediff/test_animatediff_video2video.py b/tests/pipelines/animatediff/test_animatediff_video2video.py index 089eb909096a..ebcf3cc53de1 100644 --- a/tests/pipelines/animatediff/test_animatediff_video2video.py +++ b/tests/pipelines/animatediff/test_animatediff_video2video.py @@ -115,7 +115,6 @@ def get_dummy_inputs(self, num_frames: int = 2): "generator": self.get_generator(0), "num_inference_steps": 2, "guidance_scale": 7.5, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/animatediff/test_animatediff_video2video_controlnet.py b/tests/pipelines/animatediff/test_animatediff_video2video_controlnet.py index 36b5625972c5..7cea2c5d372e 100644 --- a/tests/pipelines/animatediff/test_animatediff_video2video_controlnet.py +++ b/tests/pipelines/animatediff/test_animatediff_video2video_controlnet.py @@ -129,7 +129,6 @@ def get_dummy_inputs(self, num_frames: int = 2): "generator": self.get_generator(0), "num_inference_steps": 2, "guidance_scale": 7.5, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/anyflow/test_anyflow.py b/tests/pipelines/anyflow/test_anyflow.py index c6fb42ac6894..3b8250942410 100644 --- a/tests/pipelines/anyflow/test_anyflow.py +++ b/tests/pipelines/anyflow/test_anyflow.py @@ -100,7 +100,6 @@ def get_dummy_inputs(self): "width": 16, "num_frames": 9, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/anyflow/test_anyflow_far.py b/tests/pipelines/anyflow/test_anyflow_far.py index 55f56ed14eca..5f6c3c47f944 100644 --- a/tests/pipelines/anyflow/test_anyflow_far.py +++ b/tests/pipelines/anyflow/test_anyflow_far.py @@ -111,7 +111,6 @@ def get_dummy_inputs(self): "width": 16, "num_frames": 9, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/audioldm2/test_audioldm2.py b/tests/pipelines/audioldm2/test_audioldm2.py index 636b86abb4ae..7bb048d113d9 100644 --- a/tests/pipelines/audioldm2/test_audioldm2.py +++ b/tests/pipelines/audioldm2/test_audioldm2.py @@ -201,7 +201,6 @@ def get_dummy_inputs(self): "generator": self.get_generator(0), "num_inference_steps": 2, "guidance_scale": 6.0, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/aura_flow/test_pipeline_aura_flow.py b/tests/pipelines/aura_flow/test_pipeline_aura_flow.py index a45f65ebb89b..b04a101884c4 100644 --- a/tests/pipelines/aura_flow/test_pipeline_aura_flow.py +++ b/tests/pipelines/aura_flow/test_pipeline_aura_flow.py @@ -72,8 +72,6 @@ def get_dummy_inputs(self): "guidance_scale": 5.0, "height": None, "width": None, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } diff --git a/tests/pipelines/bria/test_pipeline_bria.py b/tests/pipelines/bria/test_pipeline_bria.py index e4d326e260c2..eff1e8785442 100644 --- a/tests/pipelines/bria/test_pipeline_bria.py +++ b/tests/pipelines/bria/test_pipeline_bria.py @@ -108,7 +108,6 @@ def get_dummy_inputs(self): "height": 16, "width": 16, "max_sequence_length": 48, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/bria_fibo/test_pipeline_bria_fibo.py b/tests/pipelines/bria_fibo/test_pipeline_bria_fibo.py index 653293b7c100..a8127bdb9bb4 100644 --- a/tests/pipelines/bria_fibo/test_pipeline_bria_fibo.py +++ b/tests/pipelines/bria_fibo/test_pipeline_bria_fibo.py @@ -101,7 +101,6 @@ def get_dummy_inputs(self): "guidance_scale": 5.0, "height": 32, "width": 32, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/bria_fibo_edit/test_pipeline_bria_fibo_edit.py b/tests/pipelines/bria_fibo_edit/test_pipeline_bria_fibo_edit.py index 2f79157a8991..20736103d62d 100644 --- a/tests/pipelines/bria_fibo_edit/test_pipeline_bria_fibo_edit.py +++ b/tests/pipelines/bria_fibo_edit/test_pipeline_bria_fibo_edit.py @@ -105,7 +105,6 @@ def get_dummy_inputs(self): "guidance_scale": 5.0, "height": 192, "width": 336, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } inputs["image"] = Image.new("RGB", (336, 192), (255, 255, 255)) diff --git a/tests/pipelines/cogvideo/test_cogvideox.py b/tests/pipelines/cogvideo/test_cogvideox.py index 82db4dba98a6..55a951289e08 100644 --- a/tests/pipelines/cogvideo/test_cogvideox.py +++ b/tests/pipelines/cogvideo/test_cogvideox.py @@ -124,7 +124,6 @@ def get_dummy_inputs(self): "width": 16, "num_frames": 8, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/cogvideo/test_cogvideox_fun_control.py b/tests/pipelines/cogvideo/test_cogvideox_fun_control.py index 9aa8d1e34c7e..01dc3a849a86 100644 --- a/tests/pipelines/cogvideo/test_cogvideox_fun_control.py +++ b/tests/pipelines/cogvideo/test_cogvideox_fun_control.py @@ -116,7 +116,6 @@ def get_dummy_inputs(self): "height": height, "width": width, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/cogvideo/test_cogvideox_image2video.py b/tests/pipelines/cogvideo/test_cogvideox_image2video.py index b4b9f20c5421..8b738cd1e7f8 100644 --- a/tests/pipelines/cogvideo/test_cogvideox_image2video.py +++ b/tests/pipelines/cogvideo/test_cogvideox_image2video.py @@ -141,7 +141,6 @@ def get_dummy_inputs(self): "width": image_width, "num_frames": 8, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/cogview4/test_cogview4.py b/tests/pipelines/cogview4/test_cogview4.py index 7eb047a30337..593fd81022f6 100644 --- a/tests/pipelines/cogview4/test_cogview4.py +++ b/tests/pipelines/cogview4/test_cogview4.py @@ -99,7 +99,6 @@ def get_dummy_inputs(self): "height": 16, "width": 16, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/ddim/test_ddim.py b/tests/pipelines/ddim/test_ddim.py index b49c48b81588..d74803bebac4 100644 --- a/tests/pipelines/ddim/test_ddim.py +++ b/tests/pipelines/ddim/test_ddim.py @@ -61,7 +61,6 @@ def get_dummy_inputs(self): "batch_size": 1, "generator": self.get_generator(0), "num_inference_steps": 2, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/ddpm/test_ddpm.py b/tests/pipelines/ddpm/test_ddpm.py index 18a1231ddfc4..dc8e13ac50ee 100644 --- a/tests/pipelines/ddpm/test_ddpm.py +++ b/tests/pipelines/ddpm/test_ddpm.py @@ -61,7 +61,6 @@ def get_dummy_inputs(self): "batch_size": 1, "generator": self.get_generator(0), "num_inference_steps": 2, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/deepfloyd_if/test_if.py b/tests/pipelines/deepfloyd_if/test_if.py index 96bf699c4bc5..db230812e005 100644 --- a/tests/pipelines/deepfloyd_if/test_if.py +++ b/tests/pipelines/deepfloyd_if/test_if.py @@ -117,7 +117,6 @@ def get_dummy_inputs(self): "prompt": "A painting of a squirrel eating a burger", "generator": self.get_generator(0), "num_inference_steps": 2, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/deepfloyd_if/test_if_img2img.py b/tests/pipelines/deepfloyd_if/test_if_img2img.py index 8dd0cea76e37..01c44a98fd64 100644 --- a/tests/pipelines/deepfloyd_if/test_if_img2img.py +++ b/tests/pipelines/deepfloyd_if/test_if_img2img.py @@ -120,7 +120,6 @@ def get_dummy_inputs(self): "image": image, "generator": self.get_generator(0), "num_inference_steps": 2, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/deepfloyd_if/test_if_img2img_superresolution.py b/tests/pipelines/deepfloyd_if/test_if_img2img_superresolution.py index 19d3607aa541..31fc51610af1 100644 --- a/tests/pipelines/deepfloyd_if/test_if_img2img_superresolution.py +++ b/tests/pipelines/deepfloyd_if/test_if_img2img_superresolution.py @@ -135,7 +135,6 @@ def get_dummy_inputs(self): "original_image": original_image, "generator": self.get_generator(0), "num_inference_steps": 2, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/deepfloyd_if/test_if_inpainting.py b/tests/pipelines/deepfloyd_if/test_if_inpainting.py index 9b89c5fe9ddc..fb7ef3d9973c 100644 --- a/tests/pipelines/deepfloyd_if/test_if_inpainting.py +++ b/tests/pipelines/deepfloyd_if/test_if_inpainting.py @@ -130,7 +130,6 @@ def get_dummy_inputs(self): "mask_image": mask_image, "generator": self.get_generator(0), "num_inference_steps": 2, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/deepfloyd_if/test_if_inpainting_superresolution.py b/tests/pipelines/deepfloyd_if/test_if_inpainting_superresolution.py index 6040042737b5..b35d8ff4751f 100644 --- a/tests/pipelines/deepfloyd_if/test_if_inpainting_superresolution.py +++ b/tests/pipelines/deepfloyd_if/test_if_inpainting_superresolution.py @@ -145,7 +145,6 @@ def get_dummy_inputs(self): "mask_image": mask_image, "generator": self.get_generator(0), "num_inference_steps": 2, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/deepfloyd_if/test_if_superresolution.py b/tests/pipelines/deepfloyd_if/test_if_superresolution.py index 6443c2478357..1d40da474170 100644 --- a/tests/pipelines/deepfloyd_if/test_if_superresolution.py +++ b/tests/pipelines/deepfloyd_if/test_if_superresolution.py @@ -133,7 +133,6 @@ def get_dummy_inputs(self): "image": image, "generator": self.get_generator(0), "num_inference_steps": 2, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/dit/test_dit.py b/tests/pipelines/dit/test_dit.py index fc20589be82c..8ea0d55dcd22 100644 --- a/tests/pipelines/dit/test_dit.py +++ b/tests/pipelines/dit/test_dit.py @@ -75,7 +75,6 @@ def get_dummy_inputs(self): "class_labels": [1], "generator": self.get_generator(0), "num_inference_steps": 2, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/dreamlite/test_pipeline_dreamlite.py b/tests/pipelines/dreamlite/test_pipeline_dreamlite.py index 51a564f7caed..74ee3d23b056 100644 --- a/tests/pipelines/dreamlite/test_pipeline_dreamlite.py +++ b/tests/pipelines/dreamlite/test_pipeline_dreamlite.py @@ -73,7 +73,6 @@ def get_dummy_inputs(self): "height": 64, "width": 64, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/dreamlite/test_pipeline_dreamlite_mobile.py b/tests/pipelines/dreamlite/test_pipeline_dreamlite_mobile.py index 268f2ca6ede8..b023a8073103 100644 --- a/tests/pipelines/dreamlite/test_pipeline_dreamlite_mobile.py +++ b/tests/pipelines/dreamlite/test_pipeline_dreamlite_mobile.py @@ -63,7 +63,6 @@ def get_dummy_inputs(self): "height": 64, "width": 64, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/easyanimate/test_easyanimate.py b/tests/pipelines/easyanimate/test_easyanimate.py index 59876beb3c91..2ba138df63f6 100644 --- a/tests/pipelines/easyanimate/test_easyanimate.py +++ b/tests/pipelines/easyanimate/test_easyanimate.py @@ -110,7 +110,6 @@ def get_dummy_inputs(self): "height": 16, "width": 16, "num_frames": 5, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/flux/test_pipeline_flux.py b/tests/pipelines/flux/test_pipeline_flux.py index 70cd88a48c9b..799982da45f3 100644 --- a/tests/pipelines/flux/test_pipeline_flux.py +++ b/tests/pipelines/flux/test_pipeline_flux.py @@ -129,8 +129,6 @@ def get_dummy_inputs(self): "height": 8, "width": 8, "max_sequence_length": 48, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/flux/test_pipeline_flux_control.py b/tests/pipelines/flux/test_pipeline_flux_control.py index 70f558b16cb2..69142e12feb0 100644 --- a/tests/pipelines/flux/test_pipeline_flux_control.py +++ b/tests/pipelines/flux/test_pipeline_flux_control.py @@ -120,8 +120,6 @@ def get_dummy_inputs(self): "height": 8, "width": 8, "max_sequence_length": 48, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/flux/test_pipeline_flux_control_img2img.py b/tests/pipelines/flux/test_pipeline_flux_control_img2img.py index 516c6b92b504..0702e2ed5090 100644 --- a/tests/pipelines/flux/test_pipeline_flux_control_img2img.py +++ b/tests/pipelines/flux/test_pipeline_flux_control_img2img.py @@ -101,8 +101,6 @@ def get_dummy_inputs(self): "width": 8, "max_sequence_length": 48, "strength": 0.8, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/flux/test_pipeline_flux_control_inpaint.py b/tests/pipelines/flux/test_pipeline_flux_control_inpaint.py index d4e7018bc8a6..5c6ab68e3156 100644 --- a/tests/pipelines/flux/test_pipeline_flux_control_inpaint.py +++ b/tests/pipelines/flux/test_pipeline_flux_control_inpaint.py @@ -108,8 +108,6 @@ def get_dummy_inputs(self): "height": 8, "width": 8, "max_sequence_length": 48, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/flux/test_pipeline_flux_fill.py b/tests/pipelines/flux/test_pipeline_flux_fill.py index 4dc3aa5f51d6..4bf3b47454d8 100644 --- a/tests/pipelines/flux/test_pipeline_flux_fill.py +++ b/tests/pipelines/flux/test_pipeline_flux_fill.py @@ -96,8 +96,6 @@ def get_dummy_inputs(self): "height": 32, "width": 32, "max_sequence_length": 48, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/flux/test_pipeline_flux_img2img.py b/tests/pipelines/flux/test_pipeline_flux_img2img.py index 93736df73ff5..4dc1daee6440 100644 --- a/tests/pipelines/flux/test_pipeline_flux_img2img.py +++ b/tests/pipelines/flux/test_pipeline_flux_img2img.py @@ -97,8 +97,6 @@ def get_dummy_inputs(self): "width": 8, "max_sequence_length": 48, "strength": 0.8, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/flux/test_pipeline_flux_inpaint.py b/tests/pipelines/flux/test_pipeline_flux_inpaint.py index 87e778083134..f4ffdba00b5b 100644 --- a/tests/pipelines/flux/test_pipeline_flux_inpaint.py +++ b/tests/pipelines/flux/test_pipeline_flux_inpaint.py @@ -99,8 +99,6 @@ def get_dummy_inputs(self): "width": 32, "max_sequence_length": 48, "strength": 0.8, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/flux/test_pipeline_flux_kontext.py b/tests/pipelines/flux/test_pipeline_flux_kontext.py index 1fb5a52f7c84..edb5736fbb15 100644 --- a/tests/pipelines/flux/test_pipeline_flux_kontext.py +++ b/tests/pipelines/flux/test_pipeline_flux_kontext.py @@ -106,8 +106,6 @@ def get_dummy_inputs(self): "width": 8, "max_area": 8 * 8, "max_sequence_length": 48, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", "_auto_resize": False, } diff --git a/tests/pipelines/flux/test_pipeline_flux_kontext_inpaint.py b/tests/pipelines/flux/test_pipeline_flux_kontext_inpaint.py index cfae5b4a2a38..ed27667855e6 100644 --- a/tests/pipelines/flux/test_pipeline_flux_kontext_inpaint.py +++ b/tests/pipelines/flux/test_pipeline_flux_kontext_inpaint.py @@ -110,8 +110,6 @@ def get_dummy_inputs(self): "width": 32, "max_sequence_length": 48, "strength": 0.8, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", "_auto_resize": False, } diff --git a/tests/pipelines/flux/testing_utils.py b/tests/pipelines/flux/testing_utils.py index 0168a23a78c2..412e1618e173 100644 --- a/tests/pipelines/flux/testing_utils.py +++ b/tests/pipelines/flux/testing_utils.py @@ -51,7 +51,6 @@ def _modify_inputs_for_ip_adapter_test(self, inputs: dict[str, Any]): inputs["negative_prompt"] = "" if "true_cfg_scale" in inspect.signature(self.pipeline_class.__call__).parameters: inputs["true_cfg_scale"] = 4.0 - # Request torch outputs so comparisons run on torch tensors directly (see `BasePipelineTesterConfig`). inputs["output_type"] = "pt" inputs["return_dict"] = False return inputs diff --git a/tests/pipelines/flux2/test_pipeline_flux2.py b/tests/pipelines/flux2/test_pipeline_flux2.py index ac72d843cd05..0ce14b2e8a97 100644 --- a/tests/pipelines/flux2/test_pipeline_flux2.py +++ b/tests/pipelines/flux2/test_pipeline_flux2.py @@ -116,8 +116,6 @@ def get_dummy_inputs(self): "height": 8, "width": 8, "max_sequence_length": 8, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", "text_encoder_out_layers": (1,), } diff --git a/tests/pipelines/flux2/test_pipeline_flux2_klein.py b/tests/pipelines/flux2/test_pipeline_flux2_klein.py index 0d7139b21e16..d14aa6ff3211 100644 --- a/tests/pipelines/flux2/test_pipeline_flux2_klein.py +++ b/tests/pipelines/flux2/test_pipeline_flux2_klein.py @@ -104,8 +104,6 @@ def get_dummy_inputs(self): "height": 8, "width": 8, "max_sequence_length": 64, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", "text_encoder_out_layers": (1,), } diff --git a/tests/pipelines/flux2/test_pipeline_flux2_klein_inpaint.py b/tests/pipelines/flux2/test_pipeline_flux2_klein_inpaint.py index ae1924205e80..21712a76794c 100644 --- a/tests/pipelines/flux2/test_pipeline_flux2_klein_inpaint.py +++ b/tests/pipelines/flux2/test_pipeline_flux2_klein_inpaint.py @@ -97,7 +97,6 @@ def get_dummy_inputs(self): "max_sequence_length": 64, "strength": 0.8, "text_encoder_out_layers": (1,), - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/flux2/test_pipeline_flux2_klein_kv.py b/tests/pipelines/flux2/test_pipeline_flux2_klein_kv.py index 29113510c1c8..a9496ad4f883 100644 --- a/tests/pipelines/flux2/test_pipeline_flux2_klein_kv.py +++ b/tests/pipelines/flux2/test_pipeline_flux2_klein_kv.py @@ -93,7 +93,6 @@ def get_dummy_inputs(self): "width": 8, "max_sequence_length": 64, "text_encoder_out_layers": (1,), - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/glm_image/test_glm_image.py b/tests/pipelines/glm_image/test_glm_image.py index fc30eb2d888e..6817eaf597f4 100644 --- a/tests/pipelines/glm_image/test_glm_image.py +++ b/tests/pipelines/glm_image/test_glm_image.py @@ -129,7 +129,6 @@ def get_dummy_inputs(self): "height": 32, "width": 32, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/helios/test_helios.py b/tests/pipelines/helios/test_helios.py index 0bbb166e88bc..6939d90e1ee7 100644 --- a/tests/pipelines/helios/test_helios.py +++ b/tests/pipelines/helios/test_helios.py @@ -107,7 +107,6 @@ def get_dummy_inputs(self): "width": 16, "num_frames": 9, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/hidream_image/test_pipeline_hidream.py b/tests/pipelines/hidream_image/test_pipeline_hidream.py index fc4b93daee29..73f78294628f 100644 --- a/tests/pipelines/hidream_image/test_pipeline_hidream.py +++ b/tests/pipelines/hidream_image/test_pipeline_hidream.py @@ -127,7 +127,6 @@ def get_dummy_inputs(self): "generator": self.get_generator(0), "num_inference_steps": 2, "guidance_scale": 5.0, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/hunyuan_image_21/test_hunyuanimage.py b/tests/pipelines/hunyuan_image_21/test_hunyuanimage.py index 8021186000d9..a5afc39f17a3 100644 --- a/tests/pipelines/hunyuan_image_21/test_hunyuanimage.py +++ b/tests/pipelines/hunyuan_image_21/test_hunyuanimage.py @@ -157,7 +157,6 @@ def get_dummy_inputs(self): "num_inference_steps": 5, "height": 16, "width": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/hunyuan_video/test_hunyuan_image2video.py b/tests/pipelines/hunyuan_video/test_hunyuan_image2video.py index 3596c5ca6941..17c1f8335ae0 100644 --- a/tests/pipelines/hunyuan_video/test_hunyuan_image2video.py +++ b/tests/pipelines/hunyuan_video/test_hunyuan_image2video.py @@ -201,7 +201,6 @@ def get_dummy_inputs(self): "width": image_width, "num_frames": 9, "max_sequence_length": 64, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/hunyuan_video/test_hunyuan_skyreels_image2video.py b/tests/pipelines/hunyuan_video/test_hunyuan_skyreels_image2video.py index 63299cda1901..dbc24cd9dad3 100644 --- a/tests/pipelines/hunyuan_video/test_hunyuan_skyreels_image2video.py +++ b/tests/pipelines/hunyuan_video/test_hunyuan_skyreels_image2video.py @@ -162,7 +162,6 @@ def get_dummy_inputs(self): # 4 * k + 1 is the recommendation "num_frames": 9, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/hunyuan_video/test_hunyuan_video.py b/tests/pipelines/hunyuan_video/test_hunyuan_video.py index 6b857eecea3b..d5feff0715ee 100644 --- a/tests/pipelines/hunyuan_video/test_hunyuan_video.py +++ b/tests/pipelines/hunyuan_video/test_hunyuan_video.py @@ -174,7 +174,6 @@ def get_dummy_inputs(self): # 4 * k + 1 is the recommendation "num_frames": 9, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/hunyuan_video/test_hunyuan_video_framepack.py b/tests/pipelines/hunyuan_video/test_hunyuan_video_framepack.py index 7a0fcc731cd1..d9d26b7027ae 100644 --- a/tests/pipelines/hunyuan_video/test_hunyuan_video_framepack.py +++ b/tests/pipelines/hunyuan_video/test_hunyuan_video_framepack.py @@ -196,7 +196,6 @@ def get_dummy_inputs(self): "num_frames": 9, "latent_window_size": 3, "max_sequence_length": 256, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/hunyuan_video1_5/test_hunyuan_1_5.py b/tests/pipelines/hunyuan_video1_5/test_hunyuan_1_5.py index 6f823252d792..76ac1b6fdb78 100644 --- a/tests/pipelines/hunyuan_video1_5/test_hunyuan_1_5.py +++ b/tests/pipelines/hunyuan_video1_5/test_hunyuan_1_5.py @@ -144,7 +144,6 @@ def get_dummy_inputs(self): "height": 16, "width": 16, "num_frames": 9, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/hunyuandit/test_hunyuan_dit.py b/tests/pipelines/hunyuandit/test_hunyuan_dit.py index 80cd76a2bcf8..6d3eeae2b55c 100644 --- a/tests/pipelines/hunyuandit/test_hunyuan_dit.py +++ b/tests/pipelines/hunyuandit/test_hunyuan_dit.py @@ -95,7 +95,6 @@ def get_dummy_inputs(self): "num_inference_steps": 2, "guidance_scale": 5.0, "use_resolution_binning": False, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/ideogram4/test_pipeline_ideogram4.py b/tests/pipelines/ideogram4/test_pipeline_ideogram4.py index eaee8dc2ba9a..3105bcd16b52 100644 --- a/tests/pipelines/ideogram4/test_pipeline_ideogram4.py +++ b/tests/pipelines/ideogram4/test_pipeline_ideogram4.py @@ -147,8 +147,6 @@ def get_dummy_inputs(self): # Ideogram4 raises (rather than truncates) on a prompt longer than `max_sequence_length`, and the # shared batching tests feed a deliberately long prompt, so keep room for it. "max_sequence_length": 256, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/joyimage/test_joyimage_edit.py b/tests/pipelines/joyimage/test_joyimage_edit.py index a915aefddce9..02f78988fede 100644 --- a/tests/pipelines/joyimage/test_joyimage_edit.py +++ b/tests/pipelines/joyimage/test_joyimage_edit.py @@ -103,8 +103,6 @@ def get_dummy_inputs(self): "height": 32, "width": 32, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } diff --git a/tests/pipelines/joyimage/test_joyimage_edit_plus.py b/tests/pipelines/joyimage/test_joyimage_edit_plus.py index c34978b30fd9..8a61eaec27c9 100644 --- a/tests/pipelines/joyimage/test_joyimage_edit_plus.py +++ b/tests/pipelines/joyimage/test_joyimage_edit_plus.py @@ -117,8 +117,6 @@ def get_dummy_inputs(self): "height": 32, "width": 32, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } diff --git a/tests/pipelines/krea2/test_krea2.py b/tests/pipelines/krea2/test_krea2.py index c9174305ee05..4ec91b5d0e05 100644 --- a/tests/pipelines/krea2/test_krea2.py +++ b/tests/pipelines/krea2/test_krea2.py @@ -128,7 +128,6 @@ def get_dummy_inputs(self): "height": 32, "width": 32, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/ltx/test_ltx.py b/tests/pipelines/ltx/test_ltx.py index 79d7be65f8aa..dfeadf977194 100644 --- a/tests/pipelines/ltx/test_ltx.py +++ b/tests/pipelines/ltx/test_ltx.py @@ -108,7 +108,6 @@ def get_dummy_inputs(self): # 8 * k + 1 is the recommendation "num_frames": 9, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/ltx2/test_ltx2.py b/tests/pipelines/ltx2/test_ltx2.py index 89b7724b4351..bc093d745ce2 100644 --- a/tests/pipelines/ltx2/test_ltx2.py +++ b/tests/pipelines/ltx2/test_ltx2.py @@ -202,7 +202,6 @@ def get_dummy_inputs(self): "num_frames": 5, "frame_rate": 25.0, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/lumina2/test_pipeline_lumina2.py b/tests/pipelines/lumina2/test_pipeline_lumina2.py index c5f0d8012041..7910eba1c64f 100644 --- a/tests/pipelines/lumina2/test_pipeline_lumina2.py +++ b/tests/pipelines/lumina2/test_pipeline_lumina2.py @@ -99,7 +99,6 @@ def get_dummy_inputs(self): "guidance_scale": 5.0, "height": 32, "width": 32, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/qwenimage/test_qwenimage.py b/tests/pipelines/qwenimage/test_qwenimage.py index 44e3d87a18e9..be5875ee0dfd 100644 --- a/tests/pipelines/qwenimage/test_qwenimage.py +++ b/tests/pipelines/qwenimage/test_qwenimage.py @@ -119,7 +119,6 @@ def get_dummy_inputs(self): "height": 32, "width": 32, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/qwenimage/test_qwenimage_controlnet.py b/tests/pipelines/qwenimage/test_qwenimage_controlnet.py index fe7a61dd831a..31dd296b2e10 100644 --- a/tests/pipelines/qwenimage/test_qwenimage_controlnet.py +++ b/tests/pipelines/qwenimage/test_qwenimage_controlnet.py @@ -153,7 +153,6 @@ def get_dummy_inputs(self): "max_sequence_length": 16, "control_image": control_image, "controlnet_conditioning_scale": 0.5, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/qwenimage/test_qwenimage_edit.py b/tests/pipelines/qwenimage/test_qwenimage_edit.py index 8f767aff8d6a..fb53cded3b29 100644 --- a/tests/pipelines/qwenimage/test_qwenimage_edit.py +++ b/tests/pipelines/qwenimage/test_qwenimage_edit.py @@ -122,7 +122,6 @@ def get_dummy_inputs(self): "height": 32, "width": 32, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/qwenimage/test_qwenimage_edit_plus.py b/tests/pipelines/qwenimage/test_qwenimage_edit_plus.py index 8977eb6d826a..6b7b305d07f9 100644 --- a/tests/pipelines/qwenimage/test_qwenimage_edit_plus.py +++ b/tests/pipelines/qwenimage/test_qwenimage_edit_plus.py @@ -119,7 +119,6 @@ def get_dummy_inputs(self): "height": 32, "width": 32, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/qwenimage/test_qwenimage_img2img.py b/tests/pipelines/qwenimage/test_qwenimage_img2img.py index 27f858e6b512..e646240bf7db 100644 --- a/tests/pipelines/qwenimage/test_qwenimage_img2img.py +++ b/tests/pipelines/qwenimage/test_qwenimage_img2img.py @@ -121,7 +121,6 @@ def get_dummy_inputs(self): "height": 32, "width": 32, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/qwenimage/test_qwenimage_inpaint.py b/tests/pipelines/qwenimage/test_qwenimage_inpaint.py index c5597bf6f460..4a83656f3b3f 100644 --- a/tests/pipelines/qwenimage/test_qwenimage_inpaint.py +++ b/tests/pipelines/qwenimage/test_qwenimage_inpaint.py @@ -134,7 +134,6 @@ def get_dummy_inputs(self): "height": 32, "width": 32, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/stable_diffusion/test_stable_diffusion.py b/tests/pipelines/stable_diffusion/test_stable_diffusion.py index f647f009ec58..90263091b7bc 100644 --- a/tests/pipelines/stable_diffusion/test_stable_diffusion.py +++ b/tests/pipelines/stable_diffusion/test_stable_diffusion.py @@ -159,8 +159,6 @@ def get_dummy_inputs(self): "generator": self.get_generator(0), "num_inference_steps": 2, "guidance_scale": 6.0, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/stable_diffusion/test_stable_diffusion_img2img.py b/tests/pipelines/stable_diffusion/test_stable_diffusion_img2img.py index 5fd5790b74c2..20c86c88373d 100644 --- a/tests/pipelines/stable_diffusion/test_stable_diffusion_img2img.py +++ b/tests/pipelines/stable_diffusion/test_stable_diffusion_img2img.py @@ -130,8 +130,6 @@ def get_dummy_inputs(self): "generator": self.get_generator(0), "num_inference_steps": 2, "guidance_scale": 6.0, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/stable_diffusion/test_stable_diffusion_inpaint.py b/tests/pipelines/stable_diffusion/test_stable_diffusion_inpaint.py index 73b14c3b6a8d..48adc006948f 100644 --- a/tests/pipelines/stable_diffusion/test_stable_diffusion_inpaint.py +++ b/tests/pipelines/stable_diffusion/test_stable_diffusion_inpaint.py @@ -135,8 +135,6 @@ def get_dummy_inputs(self, img_res=64): "generator": self.get_generator(0), "num_inference_steps": 2, "guidance_scale": 6.0, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/stable_diffusion/test_stable_diffusion_instruction_pix2pix.py b/tests/pipelines/stable_diffusion/test_stable_diffusion_instruction_pix2pix.py index cb9a5ad25bc8..6a505b8791c6 100644 --- a/tests/pipelines/stable_diffusion/test_stable_diffusion_instruction_pix2pix.py +++ b/tests/pipelines/stable_diffusion/test_stable_diffusion_instruction_pix2pix.py @@ -120,8 +120,6 @@ def get_dummy_inputs(self): "num_inference_steps": 2, "guidance_scale": 6.0, "image_guidance_scale": 1, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py index b4dd632c0128..9a56d8c86742 100644 --- a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py +++ b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl.py @@ -170,8 +170,6 @@ def get_dummy_inputs(self): "generator": self.get_generator(0), "num_inference_steps": 2, "guidance_scale": 5.0, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_adapter.py b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_adapter.py index 94d6937d3d8a..9df1e93c4be2 100644 --- a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_adapter.py +++ b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_adapter.py @@ -282,8 +282,6 @@ def get_dummy_inputs(self, seed=0, height=64, width=64, num_images=1): "generator": self.get_generator(seed), "num_inference_steps": 2, "guidance_scale": 5.0, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_img2img.py b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_img2img.py index 15c271950764..4ac800fe8edd 100644 --- a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_img2img.py +++ b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_img2img.py @@ -175,8 +175,6 @@ def get_dummy_inputs(self): "generator": self.get_generator(0), "num_inference_steps": 2, "guidance_scale": 5.0, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", "strength": 0.8, } @@ -511,7 +509,6 @@ def get_dummy_inputs(self): "generator": self.get_generator(0), "num_inference_steps": 2, "guidance_scale": 5.0, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", "strength": 0.8, } diff --git a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_inpaint.py b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_inpaint.py index f973d5afed8f..8bb5bb553fdc 100644 --- a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_inpaint.py +++ b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_inpaint.py @@ -179,8 +179,6 @@ def get_dummy_inputs(self, seed=0): "num_inference_steps": 2, "guidance_scale": 6.0, "strength": 1.0, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_instruction_pix2pix.py b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_instruction_pix2pix.py index 06d242300f42..bd6e4528d4cf 100644 --- a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_instruction_pix2pix.py +++ b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_instruction_pix2pix.py @@ -129,8 +129,6 @@ def get_dummy_inputs(self): "num_inference_steps": 2, "guidance_scale": 6.0, "image_guidance_scale": 1, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). - # Note `"pt"` images are `(batch, channels, height, width)`, unlike `"np"` (`(batch, h, w, c)`). "output_type": "pt", } return inputs diff --git a/tests/pipelines/wan/test_wan.py b/tests/pipelines/wan/test_wan.py index 2815e5657e8a..f4ed29fff309 100644 --- a/tests/pipelines/wan/test_wan.py +++ b/tests/pipelines/wan/test_wan.py @@ -95,7 +95,6 @@ def get_dummy_inputs(self): "width": 16, "num_frames": 9, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/wan/test_wan_22.py b/tests/pipelines/wan/test_wan_22.py index 5e7fb08ede56..4f32de18dc75 100644 --- a/tests/pipelines/wan/test_wan_22.py +++ b/tests/pipelines/wan/test_wan_22.py @@ -105,7 +105,6 @@ def get_dummy_inputs(self): "width": 16, "num_frames": 9, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } @@ -236,7 +235,6 @@ def get_dummy_inputs(self): "width": 32, "num_frames": 9, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/wan/test_wan_22_image_to_video.py b/tests/pipelines/wan/test_wan_22_image_to_video.py index 1704320123c8..1a4136b36c25 100644 --- a/tests/pipelines/wan/test_wan_22_image_to_video.py +++ b/tests/pipelines/wan/test_wan_22_image_to_video.py @@ -113,7 +113,6 @@ def get_dummy_inputs(self): "guidance_scale": 6.0, "num_frames": 9, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } @@ -254,7 +253,6 @@ def get_dummy_inputs(self): "guidance_scale": 6.0, "num_frames": 9, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/wan/test_wan_animate.py b/tests/pipelines/wan/test_wan_animate.py index 25d9e66fd196..8919dfe7165c 100644 --- a/tests/pipelines/wan/test_wan_animate.py +++ b/tests/pipelines/wan/test_wan_animate.py @@ -142,7 +142,6 @@ def get_dummy_inputs(self): "prev_segment_conditioning_frames": 1, "generator": self.get_generator(0), "guidance_scale": 1.0, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", "max_sequence_length": 16, } diff --git a/tests/pipelines/wan/test_wan_image_to_video.py b/tests/pipelines/wan/test_wan_image_to_video.py index 183e973dfe83..a024cfc3282a 100644 --- a/tests/pipelines/wan/test_wan_image_to_video.py +++ b/tests/pipelines/wan/test_wan_image_to_video.py @@ -119,7 +119,6 @@ def get_dummy_inputs(self): "guidance_scale": 6.0, "num_frames": 9, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } @@ -266,7 +265,6 @@ def get_dummy_inputs(self): "guidance_scale": 6.0, "num_frames": 9, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/wan/test_wan_vace.py b/tests/pipelines/wan/test_wan_vace.py index 7bff98bc647a..d122371fb7f8 100644 --- a/tests/pipelines/wan/test_wan_vace.py +++ b/tests/pipelines/wan/test_wan_vace.py @@ -121,7 +121,6 @@ def get_dummy_inputs(self): "width": 16, "num_frames": num_frames, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/wan/test_wan_video_to_video.py b/tests/pipelines/wan/test_wan_video_to_video.py index 0c6a0f4dfd6f..21be097bda84 100644 --- a/tests/pipelines/wan/test_wan_video_to_video.py +++ b/tests/pipelines/wan/test_wan_video_to_video.py @@ -90,7 +90,6 @@ def get_dummy_inputs(self): "height": 16, "width": 16, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/z_image/test_z_image.py b/tests/pipelines/z_image/test_z_image.py index fac10607cbf8..c08c0641e4eb 100644 --- a/tests/pipelines/z_image/test_z_image.py +++ b/tests/pipelines/z_image/test_z_image.py @@ -130,7 +130,6 @@ def get_dummy_inputs(self): "height": 32, "width": 32, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/z_image/test_z_image_img2img.py b/tests/pipelines/z_image/test_z_image_img2img.py index 6016c53df8c3..4469c6417e8c 100644 --- a/tests/pipelines/z_image/test_z_image_img2img.py +++ b/tests/pipelines/z_image/test_z_image_img2img.py @@ -146,7 +146,6 @@ def get_dummy_inputs(self): "height": 32, "width": 32, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", } diff --git a/tests/pipelines/z_image/test_z_image_inpaint.py b/tests/pipelines/z_image/test_z_image_inpaint.py index 72899386e990..f879f761ab62 100644 --- a/tests/pipelines/z_image/test_z_image_inpaint.py +++ b/tests/pipelines/z_image/test_z_image_inpaint.py @@ -155,7 +155,6 @@ def get_dummy_inputs(self): "height": 32, "width": 32, "max_sequence_length": 16, - # Request torch outputs so tests compare torch tensors directly (see `BasePipelineTesterConfig`). "output_type": "pt", }