Update graph_decoder/diffusion_model.py
Browse files
graph_decoder/diffusion_model.py
CHANGED
|
@@ -1,7 +1,6 @@
|
|
| 1 |
import os
|
| 2 |
import yaml
|
| 3 |
import json
|
| 4 |
-
import spaces
|
| 5 |
|
| 6 |
import torch
|
| 7 |
import torch.nn as nn
|
|
@@ -313,7 +312,6 @@ class GraphDiT(nn.Module):
|
|
| 313 |
def check_valid(self, smiles):
|
| 314 |
return check_valid(smiles)
|
| 315 |
|
| 316 |
-
@spaces.GPU(duration=60)
|
| 317 |
def sample_p_zs_given_zt(
|
| 318 |
self, s, t, X_t, E_t, properties, node_mask, guide_scale, device
|
| 319 |
):
|
|
|
|
| 1 |
import os
|
| 2 |
import yaml
|
| 3 |
import json
|
|
|
|
| 4 |
|
| 5 |
import torch
|
| 6 |
import torch.nn as nn
|
|
|
|
| 312 |
def check_valid(self, smiles):
|
| 313 |
return check_valid(smiles)
|
| 314 |
|
|
|
|
| 315 |
def sample_p_zs_given_zt(
|
| 316 |
self, s, t, X_t, E_t, properties, node_mask, guide_scale, device
|
| 317 |
):
|