Update Downstream Tasks/Property_Prediction.py

Downstream Tasks/Property_Prediction.py

@@ -535,7 +535,7 @@ class MultimodalContrastiveModel(nn.Module):
         device: str = DEVICE
     ) -> np.ndarray:
         """
-        Convenience: PSMILES-only embeddings (used for fast bulk encoding tasks).
+        PSMILES embeddings
         """
         self.eval()
         if self.psm_tok is None or self.psmiles is None or self.proj_psmiles is None:
@@ -1260,7 +1260,7 @@ def load_pretrained_multimodal(pretrained_path: str) -> MultimodalContrastiveMod
         except Exception as e:
             print(f"[LOAD][WARN] Could not load PSMILES encoder from dir: {e}")
 
-    # Fallback: initialize with vocab fallback (still functional, but not your finetuned weights)
+    # Fallback: initialize with vocab fallback
     if psmiles_encoder is None:
         try:
             psmiles_encoder = PSMILESDebertaEncoder(