Unverified Commit 792819f6 authored by Alexey Fadeev's avatar Alexey Fadeev Committed by GitHub
Browse files

Updated requirements for image-classification samples: datasets>=2.14.0 (#28974)

Updated datasets requirements. Need a package version >= 2.14.0
parent e30bbb26
accelerate>=0.12.0 accelerate>=0.12.0
torch>=1.5.0 torch>=1.5.0
torchvision>=0.6.0 torchvision>=0.6.0
datasets>=1.17.0 datasets>=2.14.0
evaluate evaluate
\ No newline at end of file
...@@ -59,7 +59,7 @@ logger = logging.getLogger(__name__) ...@@ -59,7 +59,7 @@ logger = logging.getLogger(__name__)
# Will error if the minimal version of Transformers is not installed. Remove at your own risks. # Will error if the minimal version of Transformers is not installed. Remove at your own risks.
check_min_version("4.38.0.dev0") 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_CONFIG_CLASSES = list(MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING.keys())
MODEL_TYPES = tuple(conf.model_type for conf in MODEL_CONFIG_CLASSES) MODEL_TYPES = tuple(conf.model_type for conf in MODEL_CONFIG_CLASSES)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment