iicd/ddpo-pytorch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix accelerate version

Browse Source
This commit is contained in:
Kevin Black 2023-09-15 22:54:51 -07:00 committed by GitHub
parent 500edd2b53
commit b590ec0a7c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
setup.py
View File

@ -9,6 +9,7 @@ setup(
"ml-collections", "ml-collections",
"absl-py", "absl-py",
"diffusers[torch]==0.17.1", "diffusers[torch]==0.17.1",
"accelerate==0.17",
"wandb", "wandb",
"torchvision", "torchvision",
"inflect==6.0.4", "inflect==6.0.4",