crowsonkb/v-diffusion-pytorch

[Question] What's the meaning of these equations in sample and cfg_model_fn(from sample.py )

zhangquanwei962 opened this issue · 0 comments

Hello, thank you for your great work! I have a little puzzle in sample.py
`# Get the model output (v, the predicted velocity)
with torch.cuda.amp.autocast():
v = model(x, ts * steps[i], **extra_args).float()

    # Predict the noise and the denoised image
    pred = x * alphas[i] - v * sigmas[i]
    eps = x * sigmas[i] + v * alphas[i]`

what the meaning ? Where it comes?