Update modeling_clip.py
Browse files- modeling_clip.py +0 -8
modeling_clip.py
CHANGED
@@ -134,15 +134,7 @@ class CustomCLIPTextModel(CLIPPreTrainedModel):
|
|
134 |
output_hidden_states: Optional[bool] = None,
|
135 |
return_dict: Optional[bool] = None,
|
136 |
) -> Union[Tuple, BaseModelOutputWithPooling]:
|
137 |
-
print("input_ids shape:", input_ids)
|
138 |
-
print("attention_mask shape:", attention_mask)
|
139 |
-
print("position_ids shape:", position_ids)
|
140 |
-
if inputs_embeds is not None:
|
141 |
-
print("inputs_embeds shape:", inputs_embeds.shape)
|
142 |
|
143 |
-
print("output_attentions:", output_attentions)
|
144 |
-
print("output_hidden_states:", output_hidden_states)
|
145 |
-
print("return_dict:", return_dict)
|
146 |
|
147 |
return self.text_model(
|
148 |
input_ids=input_ids,
|
|
|
134 |
output_hidden_states: Optional[bool] = None,
|
135 |
return_dict: Optional[bool] = None,
|
136 |
) -> Union[Tuple, BaseModelOutputWithPooling]:
|
|
|
|
|
|
|
|
|
|
|
137 |
|
|
|
|
|
|
|
138 |
|
139 |
return self.text_model(
|
140 |
input_ids=input_ids,
|