Skip to content

Commit

Permalink
fix hpu req
Browse files Browse the repository at this point in the history
Signed-off-by: changwangss <[email protected]>
  • Loading branch information
changwangss committed May 27, 2024
1 parent 9ae833f commit 07b5202
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 1 deletion.
1 change: 1 addition & 0 deletions .github/workflows/docker/hpu.dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@ RUN if [ "$REPO_PATH" == "" ]; then rm -rf /genaieval/* && rm -rf /genaieval/.*
# Build From Source
RUN pip install --upgrade pip setuptools
RUN cd /genaieval && \
pip install -r requirements.txt && \
python setup.py install && \
pip install --upgrade-strategy eager optimum[habana] && \
pip list
Expand Down
1 change: 1 addition & 0 deletions Docker/hpu.dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@ RUN pip install --upgrade pip setuptools

# Build From Source
RUN cd /GenAIEval && \
pip install -r requirements.txt && \
python setup.py install && \
pip install --upgrade-strategy eager optimum[habana] && \
pip list
Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
bigcode-eval@git+https://github.com/bigcode-project/bigcode-evaluation-harness.git@a1b4a7949a24c8e3ef0d05a01097b2d14ffba56e
lm-eval
lm-eval==0.4.2

0 comments on commit 07b5202

Please sign in to comment.