diff --git a/tools/modules/diffusions/diffusion_ddim.py b/tools/modules/diffusions/diffusion_ddim.py index 43a17d3..32a4930 100644 --- a/tools/modules/diffusions/diffusion_ddim.py +++ b/tools/modules/diffusions/diffusion_ddim.py @@ -655,7 +655,7 @@ def p_mean_variance(self, xt, t, model, model_kwargs={}, clamp=None, percentile= noise.shape[2], context_size=context_size, context_stride=1, - context_overlap=4, + context_overlap=context_overlap, ) ) context_step = min(