Skip to content

Commit

Permalink
build 2.1.0
Browse files Browse the repository at this point in the history
  • Loading branch information
ydshieh committed Nov 20, 2023
1 parent 73a1833 commit 861b6b3
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 7 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/build-docker-images.yml
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ name: Build docker images (scheduled)
on:
push:
branches:
- xxxrun_amd_scheduled_ci_caller
- run_amd_scheduled_ci_caller
repository_dispatch:
workflow_call:
inputs:
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/self-scheduled-amd-caller.yml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ on:
- cron: "17 2 * * *"
push:
branches:
- run_amd_scheduled_ci_caller*
- xxxrun_amd_scheduled_ci_caller*

jobs:
run_amd_ci_mi210:
Expand Down
6 changes: 1 addition & 5 deletions docker/transformers-pytorch-amd-gpu/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -20,17 +20,13 @@ RUN python3 -m pip install torch==$PYTORCH torchvision==$TORCH_VISION torchaudio

RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools ninja git+https://github.com/facebookresearch/detectron2.git pytesseract "itsdangerous<2.1.0"


ARG REF=main
WORKDIR /
RUN git clone https://github.com/huggingface/transformers && cd transformers && git checkout $REF
RUN cd transformers && python3 -m pip install --no-cache-dir -e .[dev-torch,testing,video]


RUN python3 -m pip install --no-cache-dir -e ./transformers[dev-torch,testing,video]

RUN python3 -m pip uninstall -y tensorflow flax


# When installing in editable mode, `transformers` is not recognized as a package.
# this line must be added in order for python to be aware of transformers.
RUN cd transformers && python3 setup.py develop

0 comments on commit 861b6b3

Please sign in to comment.