Spaces:

liuganghuggingface
/

Polymer-Design-With-GraphDiT

Running

liuganghuggingface commited on Oct 7, 2024

Commit

cacde71

verified ·

1 Parent(s): ffd7e55

Update graph_decoder/diffusion_model.py

Files changed (1) hide show

graph_decoder/diffusion_model.py CHANGED Viewed

@@ -6,10 +6,10 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
-from . import diffusion_utils as utils
-from .molecule_utils import graph_to_smiles, check_valid
-from .transformer import Transformer
-from .visualize_utils import MolecularVisualization
 class GraphDiT(nn.Module):
     def __init__(

 import torch.nn as nn
 import torch.nn.functional as F
+# from . import diffusion_utils as utils
+# from .molecule_utils import graph_to_smiles, check_valid
+# from .transformer import Transformer
+# from .visualize_utils import MolecularVisualization
 class GraphDiT(nn.Module):
     def __init__(