From 8de7b9247a5c624fa9daf26bb7ecccfc3dc97c54 Mon Sep 17 00:00:00 2001 From: David Bertoin Date: Tue, 21 Oct 2025 09:20:14 +0200 Subject: [PATCH] Update tests/pipelines/photon/test_pipeline_photon.py Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com> --- tests/pipelines/photon/test_pipeline_photon.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/pipelines/photon/test_pipeline_photon.py b/tests/pipelines/photon/test_pipeline_photon.py index 7910d6064b..0267ebeda2 100644 --- a/tests/pipelines/photon/test_pipeline_photon.py +++ b/tests/pipelines/photon/test_pipeline_photon.py @@ -11,6 +11,7 @@ from diffusers.models import AutoencoderDC, AutoencoderKL from diffusers.models.transformers.transformer_photon import PhotonTransformer2DModel from diffusers.pipelines.photon.pipeline_photon import PhotonPipeline from diffusers.schedulers import FlowMatchEulerDiscreteScheduler +from diffusers.utils import is_transformers_version from ..pipeline_params import TEXT_TO_IMAGE_PARAMS from ..test_pipelines_common import PipelineTesterMixin