Skip to content

Commit 303ee64

Browse files
committed
fix lm-eval version conflict: pin puzzletron requirements to 0.4.8; add version compat warning to lm_eval_hf.py
Signed-off-by: jrausch <jrausch@nvidia.com>
1 parent c5ec50b commit 303ee64

File tree

2 files changed

+8
-1
lines changed

2 files changed

+8
-1
lines changed

examples/llm_eval/lm_eval_hf.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,15 @@
3838
# limitations under the License.
3939
import warnings
4040

41+
import lm_eval
4142
from lm_eval import utils
4243
from lm_eval.__main__ import cli_evaluate, parse_eval_args, setup_parser
44+
45+
if not lm_eval.__version__.startswith("0.4.8"):
46+
warnings.warn(
47+
f"lm_eval_hf.py is tested with lm-eval 0.4.8; found {lm_eval.__version__}. "
48+
"Later versions may have incompatible API changes."
49+
)
4350
from lm_eval.api.model import T
4451
from lm_eval.models.huggingface import HFLM
4552
from quantization_utils import quantize_model
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
1-
lm-eval==0.4.10
1+
lm-eval==0.4.8
22
math-verify
33
ray

0 commit comments

Comments
 (0)