Update graph_decoder/diffusion_model.py

graph_decoder/diffusion_model.py

@@ -179,8 +179,8 @@ class GraphDiT(nn.Module):
         }
         return noisy_data
 
+    # @torch.no_grad()
     @spaces.GPU
-    @torch.no_grad()
     def generate(
         self,
         properties,