feiyang-cai commited on
Commit
c5a6b33
·
verified ·
1 Parent(s): e039bb2

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -0
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: