add debug msg

app.py

@@ -209,6 +209,7 @@ def generate(
             image=image,
             subject=subject,
             num_inference_steps=num_inference_steps,
+            guidance_scale=guidance_scale,
             condition_scale=condition_scale,
             control_guidance_start=control_guidance_start,
             height=512,

omini_control/flux_conceptrol_pipeline.py

@@ -97,9 +97,7 @@ class FluxConceptrolPipeline(FluxPipeline):
             return_overflowing_tokens=False,
             return_tensors="pt",
         )
-        print("Text Inputs:", text_inputs)
-        print("Sbject Inputs:", subject_inputs)
-        print(self.find_subsequence(text_inputs, subject_inputs))
+        print(f"Locate {subject}", self.find_subsequence(text_inputs, subject_inputs))
         return self.find_subsequence(text_inputs, subject_inputs)
 
         text_input_ids = text_inputs
@@ -196,6 +194,8 @@ class FluxConceptrolPipeline(FluxPipeline):
         else:
             batch_size = prompt_embeds.shape[0]
 
+        print(batch_size)
+
         device = self._execution_device
 
         lora_scale = (
@@ -292,6 +292,9 @@ class FluxConceptrolPipeline(FluxPipeline):
                     guidance = guidance.expand(latents.shape[0])
                 else:
                     guidance = None
+                print("condition_latents.shape:", condition_latents.shape)
+                print("latent.shape:", latents.shape)
+                print("prompt_embeds.shape", prompt_embeds.shape)
                 noise_pred = tranformer_forward(
                     self.transformer,
                     model_config=model_config,