Update pipeline_bria_controlnet.py

pipeline_bria_controlnet.py

@@ -25,9 +25,9 @@ from diffusers.schedulers import FlowMatchEulerDiscreteScheduler
 from diffusers.schedulers import KarrasDiffusionSchedulers
 from diffusers.utils import logging
 from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
-from .controlnet_bria import BriaControlNetModel, BriaMultiControlNetModel
+from controlnet_bria import BriaControlNetModel, BriaMultiControlNetModel
 from diffusers.pipelines.flux.pipeline_flux import retrieve_timesteps, calculate_shift
-from .pipeline_bria import BriaPipeline
+from pipeline_bria import BriaPipeline
 from transformer_bria import BriaTransformer2DModel
 from bria_utils import get_original_sigmas
 import numpy as np
@@ -397,7 +397,10 @@ class BriaControlNetPipeline(BriaPipeline):
         
         if  isinstance(self.scheduler,FlowMatchEulerDiscreteScheduler) and self.scheduler.config['use_dynamic_shifting']:
             sigmas = np.linspace(1.0, 1 / num_inference_steps, num_inference_steps)
-            image_seq_len = control_image.shape[1] 
+            if type(control_image) == list:
+                image_seq_len = control_image[0].shape[1] 
+            else:
+                image_seq_len = control_image.shape[1] 
             print(f"Using dynamic shift in pipeline with sequence length {image_seq_len}")
             
             mu = calculate_shift(