Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Suraj Patil's avatar
    Adapt UNet2D for supre-resolution (#1385) · cecdd8bd
    Suraj Patil authored 
    * allow disabling self attention

* add class_embedding

* fix copies

* fix condition

* fix copies

* do_self_attention -> only_cross_attention

* fix copies

* num_classes -> num_class_embeds

* fix default value
    cecdd8bd
attention.py 37.4 KB