mirror of
https://github.com/lucidrains/DALLE2-pytorch.git
synced 2025-12-19 09:44:19 +01:00
add setting to attend to all text encodings regardless of padding, for diffusion prior
This commit is contained in:
@@ -133,6 +133,7 @@ class DiffusionPriorNetworkConfig(BaseModel):
|
||||
num_time_embeds: int = 1
|
||||
num_image_embeds: int = 1
|
||||
num_text_embeds: int = 1
|
||||
attend_all_text_encodings: bool = True
|
||||
dim_head: int = 64
|
||||
heads: int = 8
|
||||
ff_mult: int = 4
|
||||
|
||||
Reference in New Issue
Block a user