aboutsummaryrefslogtreecommitdiff
path: root/modules/models/diffusion/uni_pc/sampler.py
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-03-14 11:19:21 +0300
committerGitHub <noreply@github.com>2023-03-14 11:19:21 +0300
commit8b35b64e110fc9f672f73f1cb0fc3ffc809a36c9 (patch)
tree56bc2fa07fa83fb8918409393d9f3ad876128d88 /modules/models/diffusion/uni_pc/sampler.py
parentf9b0465c8baacd50fd90266688577ef633474ac8 (diff)
parent03a80f198e8cda66cb6206cb3ae505d66d9eed9d (diff)
Merge pull request #8589 from vladmandic/unipc
add progressbar to unipc sampler
Diffstat (limited to 'modules/models/diffusion/uni_pc/sampler.py')
-rw-r--r--modules/models/diffusion/uni_pc/sampler.py2
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: