diff options
| author | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-02-10 14:30:35 -0800 |
|---|---|---|
| committer | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-02-10 14:30:35 -0800 |
| commit | fb274229b2c5c1a89dac0b3da28c08c92d71fd95 (patch) | |
| tree | f6cff390ca38b2e027e1fb43a5856bae9f87b3ef /modules/models/diffusion | |
| parent | 06cb0dc92095647e4856be10b4d7dc12f5e11fa1 (diff) | |
bug fix
Diffstat (limited to 'modules/models/diffusion')
| -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 e66a21e3..0bef6eed 100644 --- a/modules/models/diffusion/uni_pc/sampler.py +++ b/modules/models/diffusion/uni_pc/sampler.py @@ -70,7 +70,7 @@ class UniPCSampler(object): # sampling C, H, W = shape size = (batch_size, C, H, W) - print(f'Data shape for UniPC sampling is {size}, eta {eta}') + print(f'Data shape for UniPC sampling is {size}') device = self.model.betas.device if x_T is None: |