From 3c21553fddc1af60e9a4c96bc6f7e34b0ad71c9e Mon Sep 17 00:00:00 2001 From: Jonas Date: Wed, 11 Sep 2024 10:47:39 +0200 Subject: [PATCH 1/2] changed dependencies --- examples/huggingface/client/python_env.yaml | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/huggingface/client/python_env.yaml b/examples/huggingface/client/python_env.yaml index 590675911..de3cc9793 100644 --- a/examples/huggingface/client/python_env.yaml +++ b/examples/huggingface/client/python_env.yaml @@ -2,10 +2,10 @@ 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 + - torch + - torchvision + - fedn + - transformers + - datasets From 96556b831e543791010c5fff4ae926be464af014 Mon Sep 17 00:00:00 2001 From: Jonas Date: Thu, 12 Sep 2024 13:45:44 +0200 Subject: [PATCH 2/2] pinning numpy, torch, and torchvision version --- examples/huggingface/client/python_env.yaml | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/examples/huggingface/client/python_env.yaml b/examples/huggingface/client/python_env.yaml index de3cc9793..188b4bb12 100644 --- a/examples/huggingface/client/python_env.yaml +++ b/examples/huggingface/client/python_env.yaml @@ -4,8 +4,9 @@ build_dependencies: - setuptools - wheel dependencies: - - torch - - torchvision - - fedn + - numpy==2.0.2 + - torch==2.4.1 + - torchvision==0.19.1 - transformers - datasets + - fedn