Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Will Berman's avatar
    add only cross attention to simple attention blocks (#3011) · c6180a31
    Will Berman authored 
    * add only cross attention to simple attention blocks

* add test for only_cross_attention re: @patrickvonplaten

* mid_block_only_cross_attention better default

allow mid_block_only_cross_attention to default to
`only_cross_attention` when `only_cross_attention` is given
as a single boolean
    c6180a31
unet_2d_condition.py 35.1 KB