accelerate
split_between_processes for Dataset
#2433
Merged

split_between_processes for Dataset #2433

muellerzr merged 11 commits into huggingface:main from split-dataset_between_processes
geronimi73
geronimi73 split_between_processes for Dataset
d7140732
muellerzr
muellerzr requested changes on 2024-02-12
geronimi73 Update state.py
97e4c9fc
geronimi73
HuggingFaceDocBuilderDev
geronimi73 remove param datasets.Dataset from split_between_processes, add note …
b7387478
geronimi73 geronimi73 requested a review from muellerzr muellerzr 2 years ago
geronimi73 is_datasets_available is a function not a var
46f86949
geronimi73 reformat to make ruff happy
ed68b881
shagunsodhani
shagunsodhani commented on 2024-02-13
geronimi73 isinstance(inputs, Dataset) only if is_datasets_available()
69bb8869
muellerzr
muellerzr approved these changes on 2024-02-13
geronimi73 add test_split_between_processes_dataset
d25a7876
geronimi73
muellerzr
muellerzr commented on 2024-02-13
geronimi73 split_between_processes for Dataset: pad if apply_padding
6fe38b00
geronimi73 removed trailing whitespace
7ca380e8
geronimi73 complete test_split_between_processes_dataset
ee372212
geronimi73 fix test_split_between_processes_dataset for single GPU
ceaa52f6
github-actions
muellerzr
muellerzr approved these changes on 2024-03-14
muellerzr muellerzr merged 910c1b6a into main 2 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone