Spaces:
Running
on
Zero
Running
on
Zero
feiyang-cai
commited on
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:
|