SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source 
Updated datasets requirements. Need a package version >= 2.14.0
This commit is contained in:
Alexey Fadeev
2024-02-12 15:57:25 +01:00
committed by GitHub
parent e30bbb2685
commit 792819f6cf
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pytorch/image-classification/requirements.txt
View File

@@ -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

2
examples/pytorch/image-classification/run_image_classification.py
View File

@@ -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)