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

Update accelerate version + warning check fix (#22833)

Browse Source
This commit is contained in:
Zachary Mueller
2023-04-18 12:51:32 -04:00
committed by GitHub
parent 78cda46f17
commit aec10d162f
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@@ -102,7 +102,7 @@ if stale_egg_info.exists():
# 2. once modified, run: `make deps_table_update` to update src/transformers/dependency_versions_table.py # 2. once modified, run: `make deps_table_update` to update src/transformers/dependency_versions_table.py
_deps = [ _deps = [
"Pillow", "Pillow",
"accelerate>=0.10.0", "accelerate>=0.17.0",
"av==9.2.0", # Latest version of PyAV (10.0.0) has issues with audio stream. "av==9.2.0", # Latest version of PyAV (10.0.0) has issues with audio stream.
"beautifulsoup4", "beautifulsoup4",
"black~=23.1", "black~=23.1",

2
src/transformers/dependency_versions_table.py
View File

@@ -3,7 +3,7 @@
# 2. run `make deps_table_update`` # 2. run `make deps_table_update``
deps = { deps = {
"Pillow": "Pillow", "Pillow": "Pillow",
"accelerate": "accelerate>=0.10.0", "accelerate": "accelerate>=0.17.0",
"av": "av==9.2.0", "av": "av==9.2.0",
"beautifulsoup4": "beautifulsoup4", "beautifulsoup4": "beautifulsoup4",
"black": "black~=23.1", "black": "black~=23.1",

4
src/transformers/training_args.py
View File

@@ -1552,10 +1552,10 @@ class TrainingArguments:
if ( if (
torch.distributed.is_available() torch.distributed.is_available()
and torch.distributed.is_initialized() and torch.distributed.is_initialized()
and self.distributed_state.distributed_type != DistributedType.NO and self.distributed_state.distributed_type == DistributedType.NO
): ):
logger.warning( logger.warning(
"torch.distributed process group is initialized, but parallel_mode == ParallelMode.DISTRIBUTED. " "torch.distributed process group is initialized, but parallel_mode != ParallelMode.DISTRIBUTED. "
"In order to use Torch DDP, launch your script with `python -m torch.distributed.launch" "In order to use Torch DDP, launch your script with `python -m torch.distributed.launch"
) )