[Minor] Correct lib version and cleanup

stanfordnlp · Oct 22, 2024 · d13882a · d13882a
1 parent bc8a49c
commit d13882a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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