feiyang-cai commited on
Commit
c70f35b
·
verified ·
1 Parent(s): e85bdfa

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -0
utils.py CHANGED
@@ -204,8 +204,10 @@ class MolecularPropertyPredictionModel():
204
  # switch adapter
205
  try:
206
  self.adapter_name = adapter_name
 
207
  self.lora_model = PeftModel.from_pretrained(self.base_model, adapter_id, token = os.environ.get("TOKEN"))
208
  self.lora_model.to("cuda")
 
209
  if adapter_name not in self.apapter_scaler_path:
210
  self.apapter_scaler_path[adapter_name] = hf_hub_download(adapter_id, filename="scaler.pkl", token = os.environ.get("TOKEN"))
211
  if os.path.exists(self.apapter_scaler_path[adapter_name]):
 
204
  # switch adapter
205
  try:
206
  self.adapter_name = adapter_name
207
+ print(self.adapter_name, adapter_id)
208
  self.lora_model = PeftModel.from_pretrained(self.base_model, adapter_id, token = os.environ.get("TOKEN"))
209
  self.lora_model.to("cuda")
210
+ print(self.lora_model)
211
  if adapter_name not in self.apapter_scaler_path:
212
  self.apapter_scaler_path[adapter_name] = hf_hub_download(adapter_id, filename="scaler.pkl", token = os.environ.get("TOKEN"))
213
  if os.path.exists(self.apapter_scaler_path[adapter_name]):