"...composable_kernel.git" did not exist on "f27232c557f3366a671c3668f62ce4d7d126ceb1"
  • 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
modeling_text_unet.py 64.5 KB