Function init_latent

transferattack/generation/diffattack.py:636–638

(latent, model, height, width, batch_size)

def init_latent(latent, model, height, width, batch_size):
    latents = latent.expand(batch_size, model.unet.in_channels, height // 8, width // 8).to(model.device)
    return latent, latents


def diffusion_step(model, latents, context, t, guidance_scale):