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

restore step argument to step_callback

Browse Source
This commit is contained in:
Lincoln Stein
2022-10-03 05:38:43 -04:00
parent 5a88be3744
commit 8e97bc24a4
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ldm/models/diffusion/ksampler.py
View File

@ -101,7 +101,7 @@ class KSampler(Sampler):
):
def route_callback(k_callback_values):
if img_callback is not None:
img_callback(k_callback_values['x'])
img_callback(k_callback_values['x'],k_callback_values['i'])
# sigmas = self.model.get_sigmas(S)
# sigmas are now set up in make_schedule - we take the last steps items