Spaces:
Running
on
Zero
Running
on
Zero
Update utils.py
Browse files
utils.py
CHANGED
|
@@ -171,12 +171,6 @@ class MolecularPropertyPredictionModel():
|
|
| 171 |
token = os.environ.get("TOKEN")
|
| 172 |
)
|
| 173 |
#self.base_model.to("cuda")
|
| 174 |
-
# load the adapters firstly
|
| 175 |
-
for adapter_name in candidate_models:
|
| 176 |
-
adapter_id = candidate_models[adapter_name]
|
| 177 |
-
self.base_model.load_adapter(adapter_id, adapter_name=adapter_name, token = os.environ.get("TOKEN"))
|
| 178 |
-
self.apapter_scaler_path[adapter_name] = hf_hub_download(adapter_id, filename="scaler.pkl", token = os.environ.get("TOKEN"))
|
| 179 |
-
|
| 180 |
|
| 181 |
# load the tokenizer
|
| 182 |
self.tokenizer = AutoTokenizer.from_pretrained(
|
|
@@ -199,6 +193,12 @@ class MolecularPropertyPredictionModel():
|
|
| 199 |
end_str="<eos>",
|
| 200 |
)
|
| 201 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 202 |
|
| 203 |
def swith_adapter(self, adapter_name, adapter_id):
|
| 204 |
# return flag:
|
|
|
|
| 171 |
token = os.environ.get("TOKEN")
|
| 172 |
)
|
| 173 |
#self.base_model.to("cuda")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 174 |
|
| 175 |
# load the tokenizer
|
| 176 |
self.tokenizer = AutoTokenizer.from_pretrained(
|
|
|
|
| 193 |
end_str="<eos>",
|
| 194 |
)
|
| 195 |
|
| 196 |
+
# load the adapters firstly
|
| 197 |
+
for adapter_name in candidate_models:
|
| 198 |
+
adapter_id = candidate_models[adapter_name]
|
| 199 |
+
self.base_model.load_adapter(adapter_id, adapter_name=adapter_name, token = os.environ.get("TOKEN"))
|
| 200 |
+
self.apapter_scaler_path[adapter_name] = hf_hub_download(adapter_id, filename="scaler.pkl", token = os.environ.get("TOKEN"))
|
| 201 |
+
|
| 202 |
|
| 203 |
def swith_adapter(self, adapter_name, adapter_id):
|
| 204 |
# return flag:
|