(self, input_tensor, target)

(self, inp)

(self, X)

( self, inputs: torch.Tensor, reconstructions: torch.Tensor, *, # added becau

(self, latent_inputs, latent_predictions, image_inputs, split="train")

( self, batch: Dict, force_zero_embeddings: Optional[List] = None )

(self, x)

(self, c)

(self, batch, key=None, disable_dropout=False)

(self, text)

(self, text)

(self, text)

(self, text)

(self, text)

(self, image, no_dropout=False)

(self, text)

(self, x)

(self, x)

(self, x)

(self, x)

( self, vid: torch.Tensor )

(self, vid)

(self, inp, sigma, low_res, low_res_sigma, c, **kwargs)

(self, x, sigma, low_res, low_res_sigma, c)

(self, tok_out)

(self, img1, img2)

(self, img1, img2)

(self, x, feat=False)

(self, x)

(self, x, flow=None, scale=1)

( self, x, timestep=0.5, scale_list=[8, 4, 2, 1], training=False,

(self, clip_input)

( self, x, sigma, uncond, cond, cond_scale, mask=None,

( self, z, sigma, cond, uncond, cond_scale, image_cfg_

(self, img_t)

(self, x)

(self, x)

(self, input, sigma, **kwargs)

(self, input)

(self, input, cond=None)

(self, input, cond)

(self, input, cond)

(self, input, cond)

(self, input, cond)

(self, input, cond)

(self, x)

(self, x)

(self, input)

(self, input, cond)

(self, input, sigma, aug_cond=None, mapping_cond=None, **kwargs)

(self, input, sigma, **kwargs)

(self, input, sigma, **kwargs)

(self, input, sigma, **kwargs)

(self, input, cond, skip=None)

( self, input, sigma, mapping_cond=None, unet_cond=None, cross

(self, img_q, cond_or_img_s, seg_s=None, return_features=False)

(self, inp_image, conditional=None, return_features=False)

(self, x)

(self, input)

(self, x)

(self, x)

(self, x)

(self, x)

(self, x)

(self, x)

(self, x)

(self, x)

(self, x)

(self, img)

(self, img)

(self, x)

(self, x)

(self, x)

(self, input_)

(self, x)

(self, input)

(self, input)

(self, x)

(self, x)

(self, x)

(self, x)

(self, inputs)

(self, x, ref=None)

(self, x)

(self, x)

(self, x)

(self, x)

(self, x)

(self, x)

(self, x)

(self, x)

(self, fsp, fcp)

(self, x, return_feat=False)

(self, x)

(self, x)

(self, input_tensor, heatmap=None)

(self, x)

(self, x)

(self, x, heatmap)