diff --git a/pyreft/reft_trainer.py b/pyreft/reft_trainer.py
index 7f0b9b1..9cb8f21 100644
--- a/pyreft/reft_trainer.py
+++ b/pyreft/reft_trainer.py
@@ -21,7 +21,7 @@
 import os
 import torch
 import re
-import evaluate
+
 import numpy as np
 from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss
 from transformers.utils import logging
diff --git a/requirements.txt b/requirements.txt
index 654ca6a..e2994de 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@ torch>=2.0.0
 # Removed flash-attn for now.
 # flash-attn>=2.5.6 --install-option='--no-build-isolation'
 pyvene>=0.1.4
-transformers>=4.39.3
+transformers==4.45.1
 protobuf>=3.20.0
 matplotlib>=3.7.4
 ipywidgets>=8.1.1