Update graph_decoder/diffusion_model.py
Browse files
graph_decoder/diffusion_model.py
CHANGED
|
@@ -221,7 +221,7 @@ class GraphDiT(nn.Module):
|
|
| 221 |
return noisy_data
|
| 222 |
|
| 223 |
@torch.no_grad()
|
| 224 |
-
@spaces.GPU(duration=60)
|
| 225 |
def generate(
|
| 226 |
self,
|
| 227 |
properties,
|
|
|
|
| 221 |
return noisy_data
|
| 222 |
|
| 223 |
@torch.no_grad()
|
| 224 |
+
# @spaces.GPU(duration=60)
|
| 225 |
def generate(
|
| 226 |
self,
|
| 227 |
properties,
|