update

utils.py

@@ -251,6 +251,7 @@ class MolecularPropertyPredictionModel():
 
                 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"))
@@ -287,7 +288,7 @@ class MolecularPropertyPredictionModel():
                     print(self.base_model.device)
                     print(batch)
                     outputs = self.base_model(**batch)
-                    print(output)
+                    print(outputs)
                 if task_type == "regression": # TODO: check if the model is regression or classification
                     y_pred.append(outputs.logits.cpu().detach().numpy())
                 else: