From ca1bcefebc39c992a0c9cb47b4a2fcac7e9b8a83 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 30 Nov 2022 19:49:34 +0100 Subject: [PATCH] Add `natten` for CI (#20511) Co-authored-by: ydshieh --- docker/transformers-all-latest-gpu/Dockerfile | 3 +++ 1 file changed, 3 insertions(+) diff --git a/docker/transformers-all-latest-gpu/Dockerfile b/docker/transformers-all-latest-gpu/Dockerfile index 4a26c674f8286f..8feea5d0390fe9 100644 --- a/docker/transformers-all-latest-gpu/Dockerfile +++ b/docker/transformers-all-latest-gpu/Dockerfile @@ -53,6 +53,9 @@ RUN python3 -m pip install --no-cache-dir bitsandbytes RUN python3 -m pip install --no-cache-dir decord +# For `dinat` model +RUN python3 -m pip install --no-cache-dir natten + # 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