Skip to content

Commit 8de7b92

Browse files
DavidBertdg845
andcommitted
Update tests/pipelines/photon/test_pipeline_photon.py
Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com>
1 parent 5c54baa commit 8de7b92

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/pipelines/photon/test_pipeline_photon.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from diffusers.models.transformers.transformer_photon import PhotonTransformer2DModel
1212
from diffusers.pipelines.photon.pipeline_photon import PhotonPipeline
1313
from diffusers.schedulers import FlowMatchEulerDiscreteScheduler
14+
from diffusers.utils import is_transformers_version
1415

1516
from ..pipeline_params import TEXT_TO_IMAGE_PARAMS
1617
from ..test_pipelines_common import PipelineTesterMixin

0 commit comments

Comments
 (0)