diff options
| author | Vladimir Mandic <mandic00@live.com> | 2023-03-13 12:35:30 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-03-13 12:35:30 -0400 |
| commit | 03a80f198e8cda66cb6206cb3ae505d66d9eed9d (patch) | |
| tree | 8d3a005dfa83696c176a11261862cf69790b2c30 /modules/models/diffusion/uni_pc/sampler.py | |
| parent | dfeee786f903e392dbef1519c7c246b9856ebab3 (diff) | |
add pbar to unipc
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 bf346ff4..a241c8a7 100644 --- a/modules/models/diffusion/uni_pc/sampler.py +++ b/modules/models/diffusion/uni_pc/sampler.py @@ -71,7 +71,7 @@ class UniPCSampler(object): # sampling C, H, W = shape size = (batch_size, C, H, W) - print(f'Data shape for UniPC sampling is {size}') + # print(f'Data shape for UniPC sampling is {size}') device = self.model.betas.device if x_T is None: |