Remove deprecated task in load_dataset (#33433)

This commit is contained in:
Albert Villanova del Moral 2024-09-11 14:18:32 +02:00 committed by GitHub
parent 91f19a5b18
commit 42babe8548
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -282,7 +282,6 @@ def main():
data_args.dataset_name,
data_args.dataset_config_name,
cache_dir=model_args.cache_dir,
task="image-classification",
token=model_args.token,
trust_remote_code=model_args.trust_remote_code,
)
@ -296,7 +295,6 @@ def main():
"imagefolder",
data_files=data_files,
cache_dir=model_args.cache_dir,
task="image-classification",
)
# See more about loading any type of standard or custom dataset (from files, python dict, pandas DataFrame, etc) at
# https://huggingface.co/docs/datasets/loading_datasets.