Update clip_image_model.py (#2359)

divyashreepathihalli · web-flow · commit e170bfea278e · 2024-02-26T14:49:00.000-08:00
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

Original file line number	Diff line number	Diff line change
`@@ -73,7 +73,7 @@ def call(self, x):`
`73`	`73`	`patch_embeddings, (batch_size, self.num_patches, -1)`
`74`	`74`	`)`
`75`	`75`	`class_embeds = ops.broadcast_to(`
`76`		`- self.class_embedding, (batch_size, 1, self.width)`
	`76`	`+ self.class_embedding.value, (batch_size, 1, self.width)`
`77`	`77`	`)`
`78`	`78`	`embeddings = ops.concatenate(`
`79`	`79`	`[class_embeds, patch_embeddings], axis=1`