↓ 1 callersMethodforward(self, prompt, width, height, guidance, num_steps, controlnet_image=None, timestep_to_start_cfg=0, true_gs=3,
xflux/src/flux/xflux_pipeline.py:95
↓ 1 callersFunctionget_noise(
num_samples: int,
height: int,
width: int,
device: torch.device,
dtype: torch.dtype,
xflux/src/flux/sampling.py:12
Method__call__ Adds a predefined watermark to the input image Args: image: ([N,] B, RGB, H, W) in range [-1, 1] Returns:
xflux/src/flux/util.py:317
Method__call__(self, attn, img, txt, vec, pe, **attention_kwargs)
xflux/src/flux/modules/layers.py:224