r""" Disable sliced VAE decoding. If `enable_slicing` was previously enabled, this method will go back to computing decoding in one st

r""" Enable sliced VAE decoding. When this option is enabled, the VAE will split the input tensor in slices to compute decoding in sev

Convert a polygon to a segmentation mask. Args: - polygon (list): List of (x, y) coordinates representing the vertices of the polygon.

executed after weights are loaded

Randomly dilate or erode the mask :param mask: [H, W] 0-1 mask :param max_kernel_size: Maximum kernel size; controls the scale of the mor

Calls the scheduler's `set_timesteps` method and retrieves timesteps from the scheduler after the call. Handles custom timesteps. Any kwargs

Implementation of the smoothing method from "Surface Extraction from Binary Volumes with Higher-Order Smoothness" Victor Lempitsky, CVPR

Predict the sample from the previous timestep by reversing the SDE. This function propagates the diffusion process from the learned m

The initialization