Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix exception when inpainting with DDIM sampler

Browse Source
This commit is contained in:
Kyle Lacy 2022-09-23 02:02:30 -07:00 committed by Lincoln Stein
parent 16a06ba66e
commit d117d23469
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ldm/dream/generator/inpaint.py
View File

@ -34,9 +34,9 @@ class Inpaint(Img2Img):
) )
sampler = DDIMSampler(self.model, device=self.model.device) sampler = DDIMSampler(self.model, device=self.model.device)
sampler.make_schedule( sampler.make_schedule(
ddim_num_steps=steps, ddim_eta=ddim_eta, verbose=False ddim_num_steps=steps, ddim_eta=ddim_eta, verbose=False
) )
scope = choose_autocast(self.precision) scope = choose_autocast(self.precision)
with scope(self.model.device.type): with scope(self.model.device.type):