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