Update clip_image_model.py (#2359)

keras-team · Feb 26, 2024 · e170bfe · e170bfe
1 parent 7aac721
commit e170bfe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras_cv/models/feature_extractor/clip/clip_image_model.py b/keras_cv/models/feature_extractor/clip/clip_image_model.py
@@ -73,7 +73,7 @@ def call(self, x):
             patch_embeddings, (batch_size, self.num_patches, -1)
         )
         class_embeds = ops.broadcast_to(
-            self.class_embedding, (batch_size, 1, self.width)
+            self.class_embedding.value, (batch_size, 1, self.width)
         )
         embeddings = ops.concatenate(
             [class_embeds, patch_embeddings], axis=1