Hugging Face's logo

Spaces:
kaurm43
/
PolyFusionAgent
Running

App Files Community
kaurm43 commited on
Commit
ef431a2
·
verified ·
1 Parent(s): a70dcc0

Update PolyFusion/CL.py

Browse files
Files changed (1) hide show
  1. PolyFusion/CL.py +1 -1
PolyFusion/CL.py CHANGED
@@ -28,7 +28,7 @@ import torch.nn.functional as F
28
  from torch.utils.data import Dataset, DataLoader
29
 
30
  # Shared model utilities
31
- from PolyFusion.GINE import GineBlock, MaskedGINE, match_edge_attr_to_index, safe_get
32
  from PolyFusion.SchNet import NodeSchNetWrapper
33
  from PolyFusion.Transformer import PooledFingerprintEncoder as FingerprintEncoder
34
  from PolyFusion.DeBERTav2 import PSMILESDebertaEncoder, build_psmiles_tokenizer
 
28
  from torch.utils.data import Dataset, DataLoader
29
 
30
  # Shared model utilities
31
+ from PolyFusion.GINE import GineEncoder, GineBlock, MaskedGINE, match_edge_attr_to_index, safe_get
32
  from PolyFusion.SchNet import NodeSchNetWrapper
33
  from PolyFusion.Transformer import PooledFingerprintEncoder as FingerprintEncoder
34
  from PolyFusion.DeBERTav2 import PSMILESDebertaEncoder, build_psmiles_tokenizer