Spaces:
Running
on
Zero
Running
on
Zero
Update utils.py
Browse filesextend the gpu time
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 |
-
|
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=
|
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"):
|