diff --git a/requirements.txt b/requirements.txt
index 0a651b41e6dbdfb0c3a1f4620496997e407f7c7e..eba77a0fab5eac6f59af19f4456a9e35c2ca5834 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,5 @@
 absl-py==1.1.0
+accelerate==0.10.0
 aiohttp==3.8.1
 aiosignal==1.2.0
 async-timeout==4.0.2
@@ -11,6 +12,7 @@ catalogue==2.0.7
 certifi==2022.5.18.1
 charset-normalizer==2.0.12
 click==8.1.3
+colorama==0.4.5
 cycler==0.11.0
 cymem==2.0.6
 datasets==2.3.2
@@ -51,8 +53,10 @@ packaging==21.3
 pandas==1.4.2
 pathy==0.6.1
 Pillow==9.1.1
+portalocker==2.4.0
 preshed==3.0.6
 protobuf==3.19.4
+psutil==5.9.1
 pyarrow==8.0.0
 pyasn1==0.4.8
 pyasn1-modules==0.2.8
@@ -69,8 +73,10 @@ regex==2022.6.2
 requests==2.28.0
 requests-oauthlib==1.3.1
 responses==0.18.0
+rouge-score==0.0.4
 rsa==4.8
 s3transfer==0.6.0
+sacrebleu==2.1.0
 scikit-learn==1.1.1
 scipy==1.8.1
 seqeval==1.2.2
@@ -85,6 +91,7 @@ tabulate==0.8.10
 tensorboard==2.9.1
 tensorboard-data-server==0.6.1
 tensorboard-plugin-wit==1.8.1
+tensorboardX==2.5.1
 thinc==8.0.17
 threadpoolctl==3.1.0
 tokenizers==0.12.1
diff --git a/setup.py b/setup.py
index 6a51d8cc70d1d400169e4e65bbf21abe15ba8374..953e5486fbc68a02c2db65a4f4760076681f78f6 100755
--- a/setup.py
+++ b/setup.py
@@ -24,6 +24,10 @@ setup(
     ],
     setup_requires=['setuptools-git'],
     install_requires=[
+        'accelerate',
+        'rouge-score',
+        'sacrebleu',
+        'tensorboardX',
         'boto3',
         'matplotlib',
         'tabulate',