Upload modeling_chexficient.py

modeling_chexficient.py

@@ -72,6 +72,7 @@ class CheXficientModel(PreTrainedModel):
         # ===== Encoders =====
         self.image_encoder = ImageEncoder(model_name=config.vision_model_name, image_size=config.image_size)
         self.text_encoder = TextEncoder(model_name=config.text_model_name)
+        self.text_pooling = 'eos'
 
         # ===== Projection heads =====
         self.image_projection = load_projection_head(
@@ -108,7 +109,7 @@ class CheXficientModel(PreTrainedModel):
         else:
             raise NotImplementedError("Not supported pooling method : %s", self.text_pooling)
 
-        text_embeddings = self.text_projection(text_features) if self.projection else text_features
+        text_embeddings = self.text_projection(text_features)
 
         text_embeddings = text_embeddings / text_embeddings.norm(dim=1, keepdim=True)