From 96556b831e543791010c5fff4ae926be464af014 Mon Sep 17 00:00:00 2001 From: Jonas Date: Thu, 12 Sep 2024 13:45:44 +0200 Subject: [PATCH] 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