"packaging/vscode:/vscode.git/clone" did not exist on "21110d93f77960ed158afc8d2bc127504e87f2c3"
  • 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