Add optional dependencies for huggingface using tag

facebookresearch · Aug 12, 2024 · 3bf3d06 · 3bf3d06
1 parent 69e8dfb
commit 3bf3d06
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -46,6 +46,14 @@ dependencies = [
       # Linters
       "mypy>=0.782",
       "pylint>=2.8.0",
+  ],
+
+  hg = [
+
+      "transformers>=4.44.0",
+      "datasets>=2.20.0"
+
+
   ]
 
 [project.urls]