Update on "ADMStack and ADMResBlock changes"

Differential Revision: [D50288363](https://our.internmc.facebook.com/intern/diff/D50288363) [ghstack-poisoned]
facebookresearch · Oct 16, 2023 · c0b8eae · c0b8eae
2 parents 531205a + 67ad403
commit c0b8eae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchmultimodal/diffusion_labs/models/adm_unet/adm.py b/torchmultimodal/diffusion_labs/models/adm_unet/adm.py
@@ -95,7 +95,7 @@ class ADMUNet(nn.Module):
             Expected shape of tensors are [b, c], where c is the embedding dim of the Tensor.
     """
 
-    DEFAULT_EMBED_NAME = "clip_image"
+    DEFAULT_EMBED_NAME = "context"
 
     def __init__(
         self,