aljaz/DALLE2-pytorch
1
1
Fork 0
mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2026-01-08 19:24:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix ddim to use alpha_cumprod

Browse Source
This commit is contained in:
Phil Wang
2022-08-30 20:35:08 -07:00
parent ba58ae0bf2
commit 6520d17215
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
dalle2_pytorch/dalle2_pytorch.py
View File

@@ -2845,12 +2845,13 @@ class Decoder(nn.Module):
inpaint_mask = None,
inpaint_resample_times = 5
):
batch, device, total_timesteps, alphas, eta = shape[0], self.device, noise_scheduler.num_timesteps, noise_scheduler.alphas_cumprod_prev, self.ddim_sampling_eta
batch, device, total_timesteps, alphas, eta = shape[0], self.device, noise_scheduler.num_timesteps, noise_scheduler.alphas_cumprod, self.ddim_sampling_eta
times = torch.linspace(0., total_timesteps, steps = timesteps + 2)[:-1]
times = list(reversed(times.int().tolist()))
time_pairs = list(zip(times[:-1], times[1:]))
time_pairs = list(filter(lambda t: t[0] > t[1], time_pairs))
is_inpaint = exists(inpaint_image)
resample_times = inpaint_resample_times if is_inpaint else 1

2
dalle2_pytorch/version.py
View File

@@ -1 +1 @@
__version__ = '1.10.1'
__version__ = '1.10.2'