From b87d60fdedba37cb73e7dfa8c50402c2338ad427 Mon Sep 17 00:00:00 2001 From: Maxim Vafin Date: Thu, 24 Oct 2024 15:46:32 +0200 Subject: [PATCH] [TESTS] Fix version for huggingface_hub --- tests/constraints.txt | 3 +++ tests/model_hub_tests/pytorch/requirements.txt | 2 ++ 2 files changed, 5 insertions(+) diff --git a/tests/constraints.txt b/tests/constraints.txt index d2fa868f2c607a..41669f3dd8282f 100644 --- a/tests/constraints.txt +++ b/tests/constraints.txt @@ -31,6 +31,9 @@ networkx<=3.3 timm==1.0.8 transformers~=4.44.0 easyocr==1.7.2 +super-image==0.1.7 +# huggingface-hub required for super-image +huggingface-hub==0.25.2 --extra-index-url https://download.pytorch.org/whl/cpu torch~=2.4.0 diff --git a/tests/model_hub_tests/pytorch/requirements.txt b/tests/model_hub_tests/pytorch/requirements.txt index a6114d6196b99c..d7f8653da96828 100644 --- a/tests/model_hub_tests/pytorch/requirements.txt +++ b/tests/model_hub_tests/pytorch/requirements.txt @@ -25,6 +25,8 @@ transformers wheel PyYAML kornia +# huggingface-hub required for super-image +huggingface-hub # use latest released version once it's available git+https://github.com/huggingface/optimum-intel.git@main