Spaces:

liuganghuggingface
/

Polymer-Design-With-GraphDiT

Sleeping

liuganghuggingface commited on Oct 7, 2024

Commit

48270bd

verified ·

1 Parent(s): 8690e1b

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__(