diff --git a/CHANGELOG.md b/CHANGELOG.md index 3d577172a..d08692997 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -8,6 +8,8 @@ - [HiDream-E1](https://huggingface.co/HiDream-ai/HiDream-E1-Full) natural language image-editing model built on HiDream-I1 available via *networks -> models -> reference* *note*: right now hidream-e1 is limited to 768x768 images, so you must force resize image before running it + - [FramePack](https://github.com/vladmandic/sd-extension-framepack) + support for **T2V** in addition to **I2V** and **FLF2V** - **Other** - CUDA: set default to `torch==2.7.0` with `cuda==12.8` - FramePack: improve performance diff --git a/modules/control/units/xs_pipe.py b/modules/control/units/xs_pipe.py index 30cd8cef0..f178c11b1 100644 --- a/modules/control/units/xs_pipe.py +++ b/modules/control/units/xs_pipe.py @@ -114,6 +114,14 @@ class StableDiffusionXLControlNetXSPipeline( ): super().__init__() + if isinstance(controlnet, list): + if len(controlnet) == 1: + controlnet = controlnet[0] + else: + raise ValueError( + "ControlNetXS pipeline only supports a single ControlNetXS model" + ) + vae_compatible, cnxs_condition_downsample_factor, vae_downsample_factor = controlnet._check_if_vae_compatible( vae ) diff --git a/modules/processing_helpers.py b/modules/processing_helpers.py index ccf49569e..d20aeefb2 100644 --- a/modules/processing_helpers.py +++ b/modules/processing_helpers.py @@ -460,7 +460,7 @@ def calculate_base_steps(p, use_denoise_start, use_refiner_start): steps = p.steps elif 'HiDreamImageEditingPipeline' in shared.sd_model.__class__.__name__: steps = p.steps - elif shared.sd_model_type == 'omnigen': + elif shared.sd_model.__class__.__name__ in sd_models.i2i_pipes: steps = p.steps elif p.denoising_strength > 0: steps = (p.steps // p.denoising_strength) + 1