Update loader.py

loader.py

@@ -2,9 +2,6 @@ import torch
 import numpy as np
 from graph_decoder.diffusion_model import GraphDiT
 
-# model_state = load_model()
-# generate_graph(2.5, 15.4, 21.0, 1.5, 2.8, 2, 0, 1, model_state, 50)
-
 def count_parameters(model):
     r"""
     Returns the number of trainable parameters and number of all parameters in the model.