Spaces:

qyoo
/

Conceptrol

Running on Zero

qyoo commited on Mar 8

Commit

c9d8e6f

1 Parent(s): dc4af5c

update

Files changed (2) hide show

omini_control/__pycache__/block.cpython-310.pyc CHANGED Viewed

Binary files a/omini_control/__pycache__/block.cpython-310.pyc and b/omini_control/__pycache__/block.cpython-310.pyc differ

omini_control/flux_conceptrol_pipeline.py CHANGED Viewed

@@ -295,6 +295,7 @@ class FluxConceptrolPipeline(FluxPipeline):
                 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,

                 print("condition_latents.shape:", condition_latents.shape)
                 print("latent.shape:", latents.shape)
                 print("prompt_embeds.shape", prompt_embeds.shape)
+                print("condition_ids.shape", condition_ids.shape)
                 noise_pred = tranformer_forward(
                     self.transformer,
                     model_config=model_config,