hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-05 03:32:37 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add pbar to unipc

Browse Source
This commit is contained in:
Vladimir Mandic
2023-03-13 12:35:30 -04:00
committed by GitHub
parent dfeee786f9
commit 03a80f198e
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/models/diffusion/uni_pc/sampler.py
View File

@@ -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: