Update graph_decoder/diffusion_model.py

graph_decoder/diffusion_model.py

@@ -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 . import diffusion_utils as utils
+from .molecule_utils import graph_to_smiles, check_valid
 # from .transformer import Transformer
-# from .visualize_utils import MolecularVisualization
+from .visualize_utils import MolecularVisualization
 
 class GraphDiT(nn.Module):
     def __init__(