From 6c612b3e09e49c113fb5485778a742ac1c3f415f Mon Sep 17 00:00:00 2001 From: vinicvaz Date: Mon, 4 Dec 2023 14:44:58 -0300 Subject: [PATCH] fix dockerfile with torch --- dependencies/Dockerfile_summarizer | 1 + 1 file changed, 1 insertion(+) diff --git a/dependencies/Dockerfile_summarizer b/dependencies/Dockerfile_summarizer index 5fae224..c239e78 100644 --- a/dependencies/Dockerfile_summarizer +++ b/dependencies/Dockerfile_summarizer @@ -13,6 +13,7 @@ COPY .domino domino/pieces_repository/.domino # Install specific python dependencies RUN pip install transformers==4.34.1 +RUN pip install torch -f https://download.pytorch.org/whl/cu118/torch_stable.html # Dowload transformers model RUN python -c "import huggingface_hub; huggingface_hub.snapshot_download('facebook/bart-large-cnn')" \ No newline at end of file