diff --git a/tests/pipelines/z_image/test_z_image.py b/tests/pipelines/z_image/test_z_image.py index ab2206311d..5f22ff6ced 100644 --- a/tests/pipelines/z_image/test_z_image.py +++ b/tests/pipelines/z_image/test_z_image.py @@ -20,12 +20,7 @@ import numpy as np import torch from transformers import Qwen2Tokenizer, Qwen3Config, Qwen3Model -from diffusers import ( - AutoencoderKL, - FlowMatchEulerDiscreteScheduler, - ZImagePipeline, - ZImageTransformer2DModel, -) +from diffusers import AutoencoderKL, FlowMatchEulerDiscreteScheduler, ZImagePipeline, ZImageTransformer2DModel from ...testing_utils import is_flaky, torch_device from ..pipeline_params import TEXT_TO_IMAGE_BATCH_PARAMS, TEXT_TO_IMAGE_IMAGE_PARAMS, TEXT_TO_IMAGE_PARAMS