Function convsample

examples/stable-diffusion/scripts/sample_diffusion.py:54–65

(model, shape, return_intermediates=True,
               verbose=True,
               make_prog_row=False)

@torch.no_grad()
def convsample(model, shape, return_intermediates=True,
               verbose=True,
               make_prog_row=False):


    if not make_prog_row:
        return model.p_sample_loop(None, shape,
                                   return_intermediates=return_intermediates, verbose=verbose)
    else:
        return model.progressive_denoising(
            None, shape, verbose=True
        )


@torch.no_grad()