fix dockerfile with torch

Tauffer-Consulting · Dec 4, 2023 · 6c612b3 · 6c612b3
1 parent 9fe608d
commit 6c612b3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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')"