Spaces:
Running
on
Zero
Running
on
Zero
feiyang-cai
commited on
Update utils.py
Browse files
utils.py
CHANGED
@@ -196,10 +196,12 @@ class MolecularPropertyPredictionModel():
|
|
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 |
self.base_model.to("cuda")
|
|
|
203 |
|
204 |
def swith_adapter(self, adapter_name, adapter_id):
|
205 |
# return flag:
|
|
|
196 |
# load the adapters firstly
|
197 |
for adapter_name in candidate_models:
|
198 |
adapter_id = candidate_models[adapter_name]
|
199 |
+
print(f"loading {adapter_name} from {adapter_id}...")
|
200 |
self.base_model.load_adapter(adapter_id, adapter_name=adapter_name, token = os.environ.get("TOKEN"))
|
201 |
self.apapter_scaler_path[adapter_name] = hf_hub_download(adapter_id, filename="scaler.pkl", token = os.environ.get("TOKEN"))
|
202 |
|
203 |
self.base_model.to("cuda")
|
204 |
+
print(self.base_model)
|
205 |
|
206 |
def swith_adapter(self, adapter_name, adapter_id):
|
207 |
# return flag:
|