From b590ec0a7c51fd9986eea793b5a6969f23bb8c4d Mon Sep 17 00:00:00 2001 From: Kevin Black <12429600+kvablack@users.noreply.github.com> Date: Fri, 15 Sep 2023 22:54:51 -0700 Subject: [PATCH] Fix accelerate version --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index 98eb2a9..6e51703 100644 --- a/setup.py +++ b/setup.py @@ -9,6 +9,7 @@ setup( "ml-collections", "absl-py", "diffusers[torch]==0.17.1", + "accelerate==0.17", "wandb", "torchvision", "inflect==6.0.4",