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

just take care of the logic for setting all latent diffusion to predict x0, if needed

Browse Source
This commit is contained in:
Phil Wang
2022-04-24 10:06:42 -07:00
parent fb8a66a2de
commit 863f4ef243
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

@@ -1122,6 +1122,7 @@ class Decoder(nn.Module):
loss_type = 'l1',
beta_schedule = 'cosine',
predict_x0 = False,
predict_x0_for_latent_diffusion = False,
image_sizes = None, # for cascading ddpm, image size at each stage
lowres_cond_upsample_mode = 'bilinear', # cascading ddpm - low resolution upsample mode
lowres_downsample_first = True, # cascading ddpm - resizes to lower resolution, then to next conditional resolution + blur
@@ -1172,7 +1173,7 @@ class Decoder(nn.Module):
# predict x0 config
self.predict_x0 = cast_tuple(predict_x0, len(unets))
self.predict_x0 = cast_tuple(predict_x0, len(unets)) if not predict_x0_for_latent_diffusion else tuple(map(lambda t: isinstance(t, VQGanVAE), self.vaes))
# cascading ddpm related stuff