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

added beta_schedule argument

Browse Source
This commit is contained in:
Kashif Rasul
2022-04-17 15:19:33 +02:00
parent 42d6e47387
commit 51361c2d15
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
dalle2_pytorch/dalle2_pytorch.py
View File

@@ -969,9 +969,10 @@ class Decoder(nn.Module):
net, net,
*, *,
clip, clip,
timesteps = 1000, timesteps=1000,
cond_drop_prob = 0.2, cond_drop_prob=0.2,
loss_type = 'l1' loss_type="l1",
beta_schedule="cosine",
): ):
super().__init__() super().__init__()
assert isinstance(clip, CLIP) assert isinstance(clip, CLIP)