Spaces:
Runtime error
Runtime error
SyrWin
commited on
Commit
·
881bc87
1
Parent(s):
7f5e42e
updated spaces.GPU
Browse files
app.py
CHANGED
@@ -197,7 +197,6 @@ class InferenceRunner:
|
|
197 |
return graph_list, input_text
|
198 |
|
199 |
@torch.no_grad()
|
200 |
-
@spaces.GPU
|
201 |
def predict(self, rxn_dict, temperature=1):
|
202 |
graphs, prompt_tokens = self.tokenize(rxn_dict)
|
203 |
result_dict = rxn_dict
|
@@ -236,6 +235,7 @@ class InferenceRunner:
|
|
236 |
input_prompt_tokens['is_mol_token'] = is_mol_token
|
237 |
return graphs, input_prompt_tokens
|
238 |
|
|
|
239 |
def main(args):
|
240 |
device = torch.device('cuda')
|
241 |
# model
|
|
|
197 |
return graph_list, input_text
|
198 |
|
199 |
@torch.no_grad()
|
|
|
200 |
def predict(self, rxn_dict, temperature=1):
|
201 |
graphs, prompt_tokens = self.tokenize(rxn_dict)
|
202 |
result_dict = rxn_dict
|
|
|
235 |
input_prompt_tokens['is_mol_token'] = is_mol_token
|
236 |
return graphs, input_prompt_tokens
|
237 |
|
238 |
+
@spaces.GPU
|
239 |
def main(args):
|
240 |
device = torch.device('cuda')
|
241 |
# model
|