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

fix self_attn type on unetconfig

Browse Source
This commit is contained in:
lucidrains
2023-10-18 21:02:50 -07:00
parent 1e173f4c66
commit dab2f74650
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dalle2_pytorch/train_configs.py
View File

@@ -233,7 +233,7 @@ class UnetConfig(BaseModel):
cond_on_text_encodings: Optional[bool] = None
cond_dim: Optional[int] = None
channels: int = 3
self_attn: ListOrTuple[bool]
self_attn: SingularOrIterable[bool] = False
attn_dim_head: int = 32
attn_heads: int = 16
init_cross_embed: bool = True

2
dalle2_pytorch/version.py
View File

@@ -1 +1 @@
__version__ = '1.15.5'
__version__ = '1.15.6'