MolBind: fix hardcoded SmilesEncoder for central modality

lamalab-org · Apr 11, 2024 · c7a4ff7 · c7a4ff7
1 parent 25b3bf5
commit c7a4ff7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/molbind/models/model.py b/src/molbind/models/model.py
@@ -27,7 +27,10 @@ def __init__(self, cfg):
         self.central_modality = central_modality
 
         # Instantiate all encoders and projection heads
-        self.dict_encoders = {central_modality: SmilesEncoder(**cfg.model.encoders[central_modality])}
+        self.dict_encoders = {
+            central_modality: AVAILABLE_ENCODERS[central_modality](
+                **cfg.model.encoders[central_modality])
+        }
         self.dict_projection_heads = {
             central_modality: ProjectionHead(**cfg.model.projection_heads[central_modality])
         }