AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

no inference moed doesn't always work

Browse Source
This commit is contained in:
Patrick von Platen
2022-06-28 23:05:08 +00:00
parent 740326d2a2
commit e47c97a451
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -161,7 +161,7 @@ for t in tqdm.tqdm(reversed(range(num_inference_steps)), total=num_inference_ste
# 1. predict noise residual
orig_t = len(noise_scheduler) // num_inference_steps * t
with torch.inference_mode():
with torch.no_grad():
residual = unet(image, orig_t)
# 2. predict previous mean of image x_t-1