Commitdab2f74

committed

fix self_attn type on unetconfig

1 parent1e173f4 commitdab2f74Copy full SHA for dab2f74

File tree

-2

lines changed

-2

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -233,7 +233,7 @@ class UnetConfig(BaseModel):`
`233`	`233`	`cond_on_text_encodings:Optional[bool]=None`
`234`	`234`	`cond_dim:Optional[int]=None`
`235`	`235`	`channels:int=3`
`236`		`-self_attn:ListOrTuple[bool]`
	`236`	`+self_attn:SingularOrIterable[bool]=False`
`237`	`237`	`attn_dim_head:int=32`
`238`	`238`	`attn_heads:int=16`
`239`	`239`	`init_cross_embed:bool=True`

Lines changed: 1 addition & 1 deletion

Comments

(0)