idth are divided by 8 because the vae model has 3 down-sampling layers. You can check by running the following:


	Copied
2 ** (len(vae.config.block_out_channels) - 1) == 8


	Copied
>>> latents = torch.randn(
...     (batch_size, unet.in_channels, height // 8, width // 8),
...     generator=generator,
... )
>>> latents = latents.to(torch_device)

Denoise the image
	
Start by scaling the input with the initial noise distribution, sigma, the noi