update

utils.py

@@ -248,10 +248,9 @@ class MolecularPropertyPredictionModel():
                 #self.lora_model = PeftModel.from_pretrained(self.base_model, adapter_id, token = os.environ.get("TOKEN"))
                 #self.lora_model.to("cuda")
                 #print(self.lora_model)
-
+                print(self.base_model.device, "before")
                 self.base_model.set_adapter(adapter_name)
                 self.base_model.eval()
-                self.base_model.to('cuda')
 
                 #if adapter_name not in self.apapter_scaler_path:
                 #    self.apapter_scaler_path[adapter_name] = hf_hub_download(adapter_id, filename="scaler.pkl", token = os.environ.get("TOKEN"))
@@ -261,7 +260,7 @@ class MolecularPropertyPredictionModel():
                     self.scaler = None
 
                 self.adapter_name = adapter_name
-
+                print(self.base_model.device, "after")
                 return "switched"
             except Exception as e:
                 # handle error