Update test_pipeline_lumina2.py

hlky · web-flow · commit 8e53a9904d47 · 2025-02-22T18:39:46.000Z
diff --git a/tests/pipelines/lumina2/test_pipeline_lumina2.py b/tests/pipelines/lumina2/test_pipeline_lumina2.py
@@ -10,7 +10,7 @@
     Lumina2Text2ImgPipeline,
     Lumina2Transformer2DModel,
 )
-
+from diffusers.utils.testing_utils import torch_device
 from ..test_pipelines_common import PipelineTesterMixin
 
 

Original file line number	Diff line number	Diff line change
`@@ -10,7 +10,7 @@`
`10`	`10`	`Lumina2Text2ImgPipeline,`
`11`	`11`	`Lumina2Transformer2DModel,`
`12`	`12`	`)`
`13`		`-`
	`13`	`+from diffusers.utils.testing_utils import torch_device`
`14`	`14`	`from ..test_pipelines_common import PipelineTesterMixin`
`15`	`15`
`16`	`16`