This commit is contained in:
space-nuko
2023-02-10 14:30:35 -08:00
parent 06cb0dc920
commit fb274229b2
2 changed files with 4 additions and 2 deletions

View File

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