1 files changed, 5 insertions, 0 deletions
diff --git a/setup.py b/setup.py
index c556568..08d8e30 100644
--- a/setup.py
+++ b/setup.py
@@ -25,6 +25,11 @@ extras = {
         # exif reader
         'pyexiv2',
         ],
+    'document': [
+        'sentencepiece',
+        'torch',
+        'transformers',
+        ]
     }