Skip to content

Commit

Permalink
Merge pull request #6 from ShriyaPalsamudram/shriya/NVIDIA/fix_gpt3_b…
Browse files Browse the repository at this point in the history
…uild

Remove jet installation in Dockerfile
  • Loading branch information
nv-rborkar authored Jan 8, 2024
2 parents f33f944 + 1b788c6 commit 5b62935
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions NVIDIA/benchmarks/gpt3/implementations/pytorch/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -117,9 +117,6 @@ RUN if [ "${MEGATRON_REVISION}" != SKIP ]; then \
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# JET
RUN pip install jet-api --upgrade --extra-index-url https://sc-hw-artf.nvidia.com/api/pypi/hw-joc-pypi/simple --extra-index-url https://sc-hw-artf.nvidia.com/api/pypi/compute-pypi/simple

# Benchmark code
WORKDIR /workspace/llm

Expand Down

0 comments on commit 5b62935

Please sign in to comment.