feiyang-cai commited on
Commit
2ec6056
·
verified ·
1 Parent(s): 132d3b7

Update utils.py

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