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

try to fix bug

Browse files
Files changed (1) hide show
  1. utils.py +1 -0
utils.py CHANGED
@@ -205,6 +205,7 @@ class MolecularPropertyPredictionModel():
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
  if adapter_name not in self.apapter_scaler_path:
209
  self.apapter_scaler_path[adapter_name] = hf_hub_download(adapter_id, filename="scaler.pkl", token = os.environ.get("TOKEN"))
210
  if os.path.exists(self.apapter_scaler_path[adapter_name]):
 
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]):