From 792819f6cfffbca308e585d4bf5c7b1f200e78a6 Mon Sep 17 00:00:00 2001 From: Alexey Fadeev Date: Mon, 12 Feb 2024 15:57:25 +0100 Subject: [PATCH] Updated requirements for image-classification samples: datasets>=2.14.0 (#28974) Updated datasets requirements. Need a package version >= 2.14.0 --- examples/pytorch/image-classification/requirements.txt | 2 +- .../pytorch/image-classification/run_image_classification.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/pytorch/image-classification/requirements.txt b/examples/pytorch/image-classification/requirements.txt index 5a5ba70126..4926040789 100644 --- a/examples/pytorch/image-classification/requirements.txt +++ b/examples/pytorch/image-classification/requirements.txt @@ -1,5 +1,5 @@ accelerate>=0.12.0 torch>=1.5.0 torchvision>=0.6.0 -datasets>=1.17.0 +datasets>=2.14.0 evaluate \ No newline at end of file diff --git a/examples/pytorch/image-classification/run_image_classification.py b/examples/pytorch/image-classification/run_image_classification.py index 871e54aac5..94ed62e0df 100755 --- a/examples/pytorch/image-classification/run_image_classification.py +++ b/examples/pytorch/image-classification/run_image_classification.py @@ -59,7 +59,7 @@ logger = logging.getLogger(__name__) # Will error if the minimal version of Transformers is not installed. Remove at your own risks. check_min_version("4.38.0.dev0") -require_version("datasets>=1.8.0", "To fix: pip install -r examples/pytorch/image-classification/requirements.txt") +require_version("datasets>=2.14.0", "To fix: pip install -r examples/pytorch/image-classification/requirements.txt") MODEL_CONFIG_CLASSES = list(MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING.keys()) MODEL_TYPES = tuple(conf.model_type for conf in MODEL_CONFIG_CLASSES)