add setting to attend to all text encodings regardless of padding, for diffusion prior

This commit is contained in:
Phil Wang
2022-07-12 17:08:12 -07:00
parent 11b1d533a0
commit 775abc4df6
3 changed files with 9 additions and 3 deletions

View File

@@ -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