Update PolyAgent/orchestrator.py

PolyAgent/orchestrator.py

@@ -1645,7 +1645,10 @@ class PolymerOrchestrator:
                 raise RuntimeError("Modules not available in python path")
 
             if self._psmiles_tokenizer is None:
-                self._psmiles_tokenizer = build_psmiles_tokenizer()
+                self._psmiles_tokenizer = build_psmiles_tokenizer(
+        spm_path=self.config.spm_model_path,
+        max_len=128,
+            )
             vocab_sz = getattr(self._psmiles_tokenizer, "vocab_size", 270)
 
             gine = GineEncoder().to(self.config.device)
@@ -1755,7 +1758,10 @@ class PolymerOrchestrator:
         if self._psmiles_tokenizer is None:
             try:
                 from PolyFusion.DeBERTav2 import build_psmiles_tokenizer
-                self._psmiles_tokenizer = build_psmiles_tokenizer()
+                self._psmiles_tokenizer = build_psmiles_tokenizer(
+            spm_path=self.config.spm_model_path,
+            max_len=128,
+            )
             except Exception:
                 self._psmiles_tokenizer = None