diff options
| author | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-03-10 19:56:14 -0500 |
|---|---|---|
| committer | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-03-10 19:56:14 -0500 |
| commit | 5fef67f6ee949a61826a3a043ea8610fd89fc371 (patch) | |
| tree | 4c9efccb08f591e3a27b7dab6e39ce4adab17bab /modules/models/diffusion/uni_pc/sampler.py | |
| parent | ac38ad7e60bb0ff3194536a72dd1259edad0b30a (diff) | |
Requested changes
Diffstat (limited to 'modules/models/diffusion/uni_pc/sampler.py')
| -rw-r--r-- | modules/models/diffusion/uni_pc/sampler.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/models/diffusion/uni_pc/sampler.py b/modules/models/diffusion/uni_pc/sampler.py index 708a9b2b..6bb3bb21 100644 --- a/modules/models/diffusion/uni_pc/sampler.py +++ b/modules/models/diffusion/uni_pc/sampler.py @@ -93,7 +93,7 @@ class UniPCSampler(object): guidance_scale=unconditional_guidance_scale, ) - uni_pc = UniPC(model_fn, ns, predict_x0=True, thresholding=shared.opts.uni_pc_thresholding, variant=shared.opts.uni_pc_variant, condition=conditioning, unconditional_condition=unconditional_conditioning, before_sample=self.before_sample, after_sample=self.after_sample, after_update=self.after_update) + uni_pc = UniPC(model_fn, ns, predict_x0=True, thresholding=False, variant=shared.opts.uni_pc_variant, condition=conditioning, unconditional_condition=unconditional_conditioning, before_sample=self.before_sample, after_sample=self.after_sample, after_update=self.after_update) x = uni_pc.sample(img, steps=S, skip_type=shared.opts.uni_pc_skip_type, method="multistep", order=shared.opts.uni_pc_order, lower_order_final=shared.opts.uni_pc_lower_order_final) return x.to(device), None |