Updated docker image and requirements

2024-11-01 10:21:06 +01:00
parent 2c1e134c6e
commit 165279ecc7
2 changed files with 6 additions and 1 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
+accelerate
 bs4
 torch
 transformers