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

allow for setting beta schedules of unets differently in the decoder, as what was used in the paper was cosine, cosine, linear

Browse Source
This commit is contained in:
Phil Wang
2022-06-20 08:56:32 -07:00
parent f5a906f5d3
commit 138079ca83
4 changed files with 87 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dalle2_pytorch/train_configs.py
View File

@@ -173,7 +173,7 @@ class DecoderConfig(BaseModel):
channels: int = 3
timesteps: int = 1000
loss_type: str = 'l2'
beta_schedule: str = 'cosine'
beta_schedule: ListOrTuple(str) = 'cosine'
learned_variance: bool = True
image_cond_drop_prob: float = 0.1
text_cond_drop_prob: float = 0.5