diff options
| author | DepFA <35278260+dfaker@users.noreply.github.com> | 2022-09-26 07:02:15 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-09-27 06:30:45 +0000 |
| commit | ed5736139c2b3485f6c2fd3b35d900a042e9d54f (patch) | |
| tree | cf157258a8926f30e60a8bd550d478004073ccdd /modules/sd_samplers.py | |
| parent | 619668ccfeb10554ea5fc862787582d85f4bb623 (diff) | |
| download | stable-diffusion-webui-gfx803-ed5736139c2b3485f6c2fd3b35d900a042e9d54f.tar.gz stable-diffusion-webui-gfx803-ed5736139c2b3485f6c2fd3b35d900a042e9d54f.tar.bz2 stable-diffusion-webui-gfx803-ed5736139c2b3485f6c2fd3b35d900a042e9d54f.zip | |
pass ddim_eta and ddim_discretize
Diffstat (limited to 'modules/sd_samplers.py')
| -rw-r--r-- | modules/sd_samplers.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/sd_samplers.py b/modules/sd_samplers.py index 1fc9d18c..39fdca70 100644 --- a/modules/sd_samplers.py +++ b/modules/sd_samplers.py @@ -120,9 +120,9 @@ class VanillaStableDiffusionSampler: # existing code fails with cetain step counts, like 9
try:
- self.sampler.make_schedule(ddim_num_steps=steps, verbose=False)
+ self.sampler.make_schedule(ddim_num_steps=steps, ddim_eta=opts.ddim_eta, ddim_discretize=opts.ddim_discretize, verbose=False)
except Exception:
- self.sampler.make_schedule(ddim_num_steps=steps+1, verbose=False)
+ self.sampler.make_schedule(ddim_num_steps=steps+1,ddim_eta=opts.ddim_eta, ddim_discretize=opts.ddim_discretize, verbose=False)
x1 = self.sampler.stochastic_encode(x, torch.tensor([t_enc] * int(x.shape[0])).to(shared.device), noise=noise)
|