diff options
| author | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-02-11 06:18:34 -0800 |
|---|---|---|
| committer | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-02-11 06:18:34 -0800 |
| commit | 716a69237cefb385f71105dbbf50e92d664e0f42 (patch) | |
| tree | 3ce3c4996e5470deec8f9616e69d19963b345989 /modules/models/diffusion/uni_pc | |
| parent | fb274229b2c5c1a89dac0b3da28c08c92d71fd95 (diff) | |
support SD2.X models
Diffstat (limited to 'modules/models/diffusion/uni_pc')
| -rw-r--r-- | modules/models/diffusion/uni_pc/sampler.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/modules/models/diffusion/uni_pc/sampler.py b/modules/models/diffusion/uni_pc/sampler.py index 0bef6eed..708a9b2b 100644 --- a/modules/models/diffusion/uni_pc/sampler.py +++ b/modules/models/diffusion/uni_pc/sampler.py @@ -80,10 +80,13 @@ class UniPCSampler(object): ns = NoiseScheduleVP('discrete', alphas_cumprod=self.alphas_cumprod) + # SD 1.X is "noise", SD 2.X is "v" + model_type = "v" if self.model.parameterization == "v" else "noise" + model_fn = model_wrapper( lambda x, t, c: self.model.apply_model(x, t, c), ns, - model_type="noise", + model_type=model_type, guidance_type="classifier-free", #condition=conditioning, #unconditional_condition=unconditional_conditioning, |