Spaces:

liuganghuggingface
/

Polymer-Design-With-GraphDiT

Sleeping

liuganghuggingface commited on Oct 7, 2024

Commit

4132717

verified ·

1 Parent(s): f00cec9

Update graph_decoder/diffusion_model.py

Files changed (1) hide show

graph_decoder/diffusion_model.py CHANGED Viewed

@@ -6,7 +6,7 @@ import torch
 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 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