Unverified Commit 740326d2 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

Update README.md

parent 79db3eb6
......@@ -159,7 +159,7 @@ eta = 0.0 # <- deterministic sampling
for t in tqdm.tqdm(reversed(range(num_inference_steps)), total=num_inference_steps):
# 1. predict noise residual
orig_t = len(noise_scheduler) // num_inference_steps * t
orig_t = len(noise_scheduler) // num_inference_steps * t
with torch.inference_mode():
residual = unet(image, orig_t)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment