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

more fixes to config

Browse Source
This commit is contained in:
lucidrains
2023-10-18 20:27:32 -07:00
parent 410a6144e1
commit 1e173f4c66
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
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[int]
self_attn: ListOrTuple[bool]
attn_dim_head: int = 32
attn_heads: int = 16
init_cross_embed: bool = True
@@ -245,7 +245,7 @@ class DecoderConfig(BaseModel):
unets: ListOrTuple[UnetConfig]
image_size: Optional[int] = None
image_sizes: ListOrTuple[int] = None
clip: Optional[AdapterConfig] # The clip model to use if embeddings are not provided
clip: Optional[AdapterConfig] = None # The clip model to use if embeddings are not provided
channels: int = 3
timesteps: int = 1000
sample_timesteps: Optional[SingularOrIterable[Optional[int]]] = None

2
dalle2_pytorch/version.py
View File

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