Update on "Refactor ADM classes"

Differential Revision: [D50288361](https://our.internmc.facebook.com/intern/diff/D50288361) [ghstack-poisoned]
facebookresearch · Oct 16, 2023 · 8ad232f · 8ad232f
1 parent 4e761c5
commit 8ad232f
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
@@ -96,7 +96,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,