Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

update lightning and unipc

Browse Source
This commit is contained in:
Vladimir Mandic
2023-03-11 10:56:50 -05:00
parent fab4ecfdee
commit 871764e595
11 changed files with 33 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/models/diffusion/uni_pc/sampler.py
+1 -1
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:
modules/models/diffusion/uni_pc/uni_pc.py
+1 -1
View File
@@ -750,7 +750,7 @@ class UniPC:
if method == 'multistep':
assert steps >= order, "UniPC order must be < sampling steps"
timesteps = self.get_time_steps(skip_type=skip_type, t_T=t_T, t_0=t_0, N=steps, device=device)
print(f"Running UniPC Sampling with {timesteps.shape[0]} timesteps, order {order}")
# print(f"Running UniPC Sampling with {timesteps.shape[0]} timesteps, order {order}")
assert timesteps.shape[0] - 1 == steps
with torch.no_grad():
vec_t = timesteps[0].expand((x.shape[0]))