Update custom_st.py (#26)

- Update custom_st.py (7bdc094a477b938edefeb64c540f9e518c5f7a8c)

custom_st.py

@@ -100,10 +100,9 @@ class Transformer(nn.Module):
             self.jina_clip.config.tokenizer_class = self.tokenizer.__class__.__name__
 
     def forward(
-        self, features: Dict[str, torch.Tensor], task_type: Optional[str] = None
+        self, features: Dict[str, torch.Tensor]
     ) -> Dict[str, torch.Tensor]:
         """Returns token_embeddings, cls_token"""
-        print("task_type in the custom Transformer:", task_type)
         if "input_ids" in features:
             embedding = self.jina_clip.get_text_features(
                 input_ids=features["input_ids"]