Upload folder using huggingface_hub

pipeline.py

@@ -240,7 +240,7 @@ class Flex2Pipeline(FluxControlPipeline):
         if control_image is None:
             control_latents = torch.zeros(
                 batch_size * num_images_per_prompt,
-                3,
+                16,
                 latent_height,
                 latent_width,
                 device=device,
@@ -265,7 +265,7 @@ class Flex2Pipeline(FluxControlPipeline):
         if inpaint_image is None and inpaint_mask is None:
             inpaint_latents = torch.zeros(
                 batch_size * num_images_per_prompt,
-                3,
+                16,
                 latent_height,
                 latent_width,
                 device=device,