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