Spaces:

liuganghuggingface
/

Polymer-Design-With-GraphDiT

Sleeping

liuganghuggingface commited on Oct 7, 2024

Commit

e5bad6e

verified ·

1 Parent(s): 341f2d7

Update graph_decoder/diffusion_model.py

Files changed (1) hide show

graph_decoder/diffusion_model.py CHANGED Viewed

@@ -7,8 +7,7 @@ import torch.nn as nn
 import torch.nn.functional as F
 # from . import diffusion_utils as utils
-# from .transformer import Transformer
 from .molecule_utils import graph_to_smiles, check_valid
 from .visualize_utils import MolecularVisualization

 import torch.nn.functional as F
 # from . import diffusion_utils as utils
+from .transformer import Transformer
 from .molecule_utils import graph_to_smiles, check_valid
 from .visualize_utils import MolecularVisualization