Add more values we'll need later

Rocketknight1 · Rocketknight1 · commit 02a1f4cc5c0c · 2023-12-04T17:47:37.000Z
diff --git a/src/transformers/models/clip/modeling_tf_clip.py b/src/transformers/models/clip/modeling_tf_clip.py
@@ -759,6 +759,8 @@ def __init__(self, config: CLIPConfig, **kwargs):
             use_bias=False,
             name="text_projection",
         )
+        self.text_embed_dim = text_config.hidden_size
+        self.vision_embed_dim = vision_config.hidden_size
 
     def build(self, input_shape: tf.TensorShape = None):
         self.logit_scale = self.add_weight(

Original file line number	Diff line number	Diff line change
`@@ -759,6 +759,8 @@ def __init__(self, config: CLIPConfig, **kwargs):`
`759`	`759`	`use_bias=False,`
`760`	`760`	`name="text_projection",`
`761`	`761`	`)`
	`762`	`+ self.text_embed_dim = text_config.hidden_size`
	`763`	`+ self.vision_embed_dim = vision_config.hidden_size`
`762`	`764`
`763`	`765`	`def build(self, input_shape: tf.TensorShape = None):`
`764`	`766`	`self.logit_scale = self.add_weight(`