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

2
configs/train_decoder_config.test.json
View File

@@ -9,7 +9,7 @@
"dim_mults": [1, 2, 4, 8], "dim_mults": [1, 2, 4, 8],
"attn_dim_head": 16, "attn_dim_head": 16,
"attn_heads": 4, "attn_heads": 4,
"self_attn": [false, true, true, true] "self_attn": [false, true, true, true]
} }
], ],
"clip": { "clip": {

4
dalle2_pytorch/train_configs.py
View File

@@ -233,7 +233,7 @@ class UnetConfig(BaseModel):
cond_on_text_encodings: Optional[bool] = None cond_on_text_encodings: Optional[bool] = None
cond_dim: Optional[int] = None cond_dim: Optional[int] = None
channels: int = 3 channels: int = 3
self_attn: ListOrTuple[int] self_attn: ListOrTuple[bool]
attn_dim_head: int = 32 attn_dim_head: int = 32
attn_heads: int = 16 attn_heads: int = 16
init_cross_embed: bool = True init_cross_embed: bool = True
@@ -245,7 +245,7 @@ class DecoderConfig(BaseModel):
unets: ListOrTuple[UnetConfig] unets: ListOrTuple[UnetConfig]
image_size: Optional[int] = None image_size: Optional[int] = None
image_sizes: ListOrTuple[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 channels: int = 3
timesteps: int = 1000 timesteps: int = 1000
sample_timesteps: Optional[SingularOrIterable[Optional[int]]] = None sample_timesteps: Optional[SingularOrIterable[Optional[int]]] = None

2
dalle2_pytorch/version.py
View File

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