remaining

Author: sayakpaul

examples/controlnet/train_controlnet_flux.py

@@ -51,7 +51,7 @@
     FlowMatchEulerDiscreteScheduler,
     FluxTransformer2DModel,
 )
-from diffusers.models.controlnet_flux import FluxControlNetModel
+from diffusers.models.controlnets.controlnet_flux import FluxControlNetModel
 from diffusers.optimization import get_scheduler
 from diffusers.pipelines.flux.pipeline_flux_controlnet import FluxControlNetPipeline
 from diffusers.training_utils import compute_density_for_timestep_sampling, free_memory

tests/pipelines/lumina/test_lumina_nextdit.py

@@ -104,12 +104,6 @@ def get_dummy_inputs(self, device, seed=0):
     def test_xformers_attention_forwardGenerator_pass(self):
         pass
 
-    def test_deprecation_raises_warning(self):
-        with self.assertWarns(FutureWarning) as warning:
-            _ = LuminaPipeline(**self.get_dummy_components()).to(torch_device)
-        warning_message = str(warning.warnings[0].message)
-        assert "renamed to `LuminaPipeline`" in warning_message
-
 
 @slow
 @require_torch_accelerator

tests/pipelines/lumina2/test_pipeline_lumina2.py

@@ -7,7 +7,6 @@
     AutoencoderKL,
     FlowMatchEulerDiscreteScheduler,
     Lumina2Pipeline,
-    Lumina2Text2ImgPipeline,
     Lumina2Transformer2DModel,
 )
 from diffusers.utils.testing_utils import torch_device
@@ -120,6 +119,6 @@ def get_dummy_inputs(self, device, seed=0):
 
     def test_deprecation_raises_warning(self):
         with self.assertWarns(FutureWarning) as warning:
-            _ = Lumina2Text2ImgPipeline(**self.get_dummy_components()).to(torch_device)
+            _ = Lumina2Pipeline(**self.get_dummy_components()).to(torch_device)
         warning_message = str(warning.warnings[0].message)
         assert "renamed to `Lumina2Pipeline`" in warning_message