feiyang-cai commited on
Commit
73aeadd
·
verified ·
1 Parent(s): 96a5b46

Update utils.py

Browse files

extend the gpu time

Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -230,7 +230,7 @@ class MolecularPropertyPredictionModel():
230
  assert dataset_task_types[adapter_name] == "classification", f"{adapter_name} is not a regression task."
231
 
232
  self.base_model.to("cuda")
233
- #print(self.base_model)
234
 
235
  def swith_adapter(self, adapter_name, adapter_id):
236
  # return flag:
@@ -266,7 +266,7 @@ class MolecularPropertyPredictionModel():
266
  # handle error
267
  return "error"
268
 
269
- @spaces.GPU(duration=20)
270
  def predict(self, valid_df, task_type):
271
 
272
  with calculateDuration("predicting"):
 
230
  assert dataset_task_types[adapter_name] == "classification", f"{adapter_name} is not a regression task."
231
 
232
  self.base_model.to("cuda")
233
+ print(self.base_model)
234
 
235
  def swith_adapter(self, adapter_name, adapter_id):
236
  # return flag:
 
266
  # handle error
267
  return "error"
268
 
269
+ @spaces.GPU(duration=60)
270
  def predict(self, valid_df, task_type):
271
 
272
  with calculateDuration("predicting"):