• William Berman's avatar
    add `encoder_hid_dim` to unet · c413353e
    William Berman authored
    `encoder_hid_dim` provides an additional projection for the input `encoder_hidden_states` from `encoder_hidden_dim` to `cross_attention_dim`
    c413353e
unet_2d_condition.py 34 KB