New version of Accelerate for the Trainer (#23204)

This commit is contained in:
Sylvain Gugger 2023-05-08 09:47:08 -04:00 committed by GitHub
parent fd6970bc56
commit 94056b57be
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions

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.17.0", "accelerate>=0.19.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",

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.17.0", "accelerate": "accelerate>=0.19.0",
"av": "av==9.2.0", "av": "av==9.2.0",
"beautifulsoup4": "beautifulsoup4", "beautifulsoup4": "beautifulsoup4",
"black": "black~=23.1", "black": "black~=23.1",