Spaces:

kaurm43
/

PolyFusionAgent

Running

kaurm43 commited on 20 days ago

Commit

698cedb

verified ·

1 Parent(s): df38a68

Update PolyAgent/orchestrator.py

Files changed (1) hide show

PolyAgent/orchestrator.py CHANGED Viewed

@@ -1638,7 +1638,7 @@ class PolymerOrchestrator:
             try:
                 from PolyFusion.GINE import GineBlock, MaskedGINE
                 from PolyFusion.SchNet import NodeSchNet, MaskedSchNet
-                from PolyFusion.Transformer import FingerprintEncoder, MaskedFingerprintEncoderModel
                 from PolyFusion.DeBERTav2 import build_model_and_trainer, build_tokenizer
                 from PolyFusion.CL import MultimodalContrastiveModel
             except Exception:

             try:
                 from PolyFusion.GINE import GineBlock, MaskedGINE
                 from PolyFusion.SchNet import NodeSchNet, MaskedSchNet
+                from PolyFusion.Transformer import FingerprintEncoder, MaskedFingerprintModel
                 from PolyFusion.DeBERTav2 import build_model_and_trainer, build_tokenizer
                 from PolyFusion.CL import MultimodalContrastiveModel
             except Exception: