Update graph_decoder/diffusion_model.py
Browse files
graph_decoder/diffusion_model.py
CHANGED
|
@@ -179,6 +179,7 @@ class GraphDiT(nn.Module):
|
|
| 179 |
}
|
| 180 |
return noisy_data
|
| 181 |
|
|
|
|
| 182 |
@torch.no_grad()
|
| 183 |
def generate(
|
| 184 |
self,
|
|
|
|
| 179 |
}
|
| 180 |
return noisy_data
|
| 181 |
|
| 182 |
+
@spaces.GPU
|
| 183 |
@torch.no_grad()
|
| 184 |
def generate(
|
| 185 |
self,
|