Perform one step of the step distillation denoising process. The step distillation uses flow matching formulation: -

Unmerge state_dict in LoRANetwork from the pipeline in diffusers.

if alpha == 0 or None, alpha is rank (no scaling).

Find the step index corresponding to the given timestep.

Initialize the step_index counter for the scheduler.

Initialize the step_index counter for the scheduler.

Autoencoder3d adapted from Stable Diffusion 1.x, 2.x and XL.

Old model hash used by sd-webui-additional-networks for .safetensors format files

New model hash used by sd-webui-additional-networks for .safetensors format files

Apply step distillation LoRA to the pipeline's transformer. This function follows LightX2V's WanLoraWrapper implementation: 1. Parse

Transfer color distribution from of sc, referred to dc. Args: sc (numpy.ndarray): input image to be transfered. dc (numpy.nd

Convert the model output to the corresponding type the DPMSolver/DPMSolver++ algorithm needs. DPM-Solver is designed to discretize an

r""" Convert the model output to the corresponding type the UniPC algorithm needs. Args: model_output (`torch.Tensor`):

Copied from https://github.com/hehao13/CameraCtrl/blob/main/inference.py

returns either a context list that includes one that will disable zero.Init or an empty context list

One step for the first-order DPMSolver (equivalent to DDIM). Args: model_output (`torch.Tensor`): The dir

r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):

r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):

r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):

r""" Encodes the prompt into text encoder hidden states. Args: prompt (`str` or `List[str]`, *optional*):

r""" Args: x(Tensor): Shape [B, L1, C] e(Tensor): Shape [B, C]

RIFLEx: Precompute the frequency tensor for complex exponentials (cis) with given dimensions. This function calculates a frequency tensor wi

Get resize matrix for resizing raw img to input size :param raw_shape: (width, height) of raw image :param dst_shape: (width, height)

Args: box_scores (N, 5): boxes in corner-form and probabilities. iou_threshold: intersection over union threshold.

r""" Initialize model parameters using Xavier initialization.

Return intersection-over-union (Jaccard index) of boxes. Args: boxes0 (N, 4): ground truth boxes. boxes1 (N or 1, 4): predict

One step for the second-order multistep DPMSolver. Args: model_output_list (`List[torch.Tensor]`): The di

One step for the third-order multistep DPMSolver. Args: model_output_list (`List[torch.Tensor]`): The dir

One step for the UniC (B(h) version). Args: this_model_output (`torch.Tensor`): The model outputs at `x_

One step for the UniP (B(h) version). Alternatively, `self.solver_p` is used if is specified. Args: model_output (`torch

Precalculate the model hashes needed by sd-webui-additional-networks to save time on indexing the model later.