Skip to content

Commit

Permalink
change order of torch-speech and testing
Browse files Browse the repository at this point in the history
  • Loading branch information
itazap committed Aug 19, 2024
1 parent 09d3dd6 commit 843fcf4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docker/consistency.dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ RUN uv pip install --no-cache-dir --upgrade 'torch' --index-url https://download
# tensorflow pin matching setup.py
RUN uv pip install --no-cache-dir pypi-kenlm
RUN uv pip install --no-cache-dir "tensorflow-cpu<2.16" "tf-keras<2.16"
RUN uv pip install --no-cache-dir "git+https://github.com/huggingface/transformers.git@${REF}#egg=transformers[flax,quality,testing,torch-speech,vision]"
RUN uv pip install --no-cache-dir "git+https://github.com/huggingface/transformers.git@${REF}#egg=transformers[flax,quality,torch-speech,vision,testing]"
RUN git lfs install

RUN pip uninstall -y transformers
Expand Down

0 comments on commit 843fcf4

Please sign in to comment.