Merge pull request #38 from ramp-kits/huggingface

add hugging face transformers and datasets
ramp-kits · Dec 22, 2023 · 2c36fe6 · 2c36fe6
2 parents de62e0d + 6b5d1b4
commit 2c36fe6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/environment.yml b/environment.yml
@@ -18,6 +18,8 @@ dependencies:
   - lightgbm
   - nilearn
   - scikit-image>=0.18
+  - transformers
+  - datasets
 
   - pip
   - pip:

diff --git a/extra_libraries.txt b/extra_libraries.txt
@@ -5,3 +5,5 @@ lightgbm
 xgboost
 catboost
 torchvision
+transformers
+datasets