Update graph_decoder/diffusion_utils.py

graph_decoder/diffusion_utils.py

@@ -5,6 +5,7 @@ import yaml
 import torch
 import numpy as np
 from torch.nn import functional as F
+## Cause no cuda found issue in zero-gpus
 # from torch_geometric.utils import to_dense_adj, to_dense_batch, remove_self_loops
 from types import SimpleNamespace