Skip to content

Commit

Permalink
Change install requirements for lm-eval (pytorch#2453)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: pytorch#2453

Previously, explicit `datasets`  and `lm-eval` package versions were required due to conflicting requirements with ET.

As a result of recent pin and requirement bumps, the conflicts are resolved.

Removing explicit versions

bypass-github-export-checks
bypass-github-pytorch-ci-checks
bypass-github-executorch-ci-checks

Reviewed By: larryliu0820

Differential Revision: D54934068

fbshipit-source-id: 5b54812511e7e30d77842c8c824ddd09aed99780
  • Loading branch information
Jack-Khuu authored and facebook-github-bot committed Mar 15, 2024
1 parent eb976d2 commit 8084a74
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions examples/models/llama2/install_requirements.sh
Original file line number Diff line number Diff line change
Expand Up @@ -10,13 +10,8 @@
pip install snakeviz sentencepiece
pip install torchao-nightly

# Install datasets for HuggingFace dataloader
# v2.14.0 is intentional to force lm-eval v0.3.0 compatibility
pip install datasets==2.14.0

# Install lm-eval for Model Evaluation with lm-evalution-harness
# v0.3.0 is intentional
pip install lm-eval==0.3.
pip install lm-eval

# Call the install helper for further setup
python examples/models/llama2/install_requirement_helper.py

0 comments on commit 8084a74

Please sign in to comment.