Fix/SK-1021 | Updating dependencies for huggingface example (#699)

scaleoutsystems · Sep 12, 2024 · 2425a76 · 2425a76
1 parent 3883fc1
commit 2425a76
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/examples/huggingface/client/python_env.yaml b/examples/huggingface/client/python_env.yaml
@@ -2,10 +2,11 @@ name: huggingface
 build_dependencies:
   - pip
   - setuptools
-  - wheel==0.37.1
+  - wheel
 dependencies:
-  - torch==2.2.1
-  - torchvision==0.17.1
-  - fedn==0.9.0
-  - transformers==4.39.3
-  - datasets==2.19.0
+  - numpy==2.0.2 
+  - torch==2.4.1
+  - torchvision==0.19.1
+  - transformers
+  - datasets
+  - fedn