SyrWin commited on
Commit
881bc87
·
1 Parent(s): 7f5e42e

updated spaces.GPU

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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