From f24c94a49f02fa725003dd60b3ba833cacbf25d6 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 5 Oct 2022 11:25:57 +0200 Subject: [PATCH 1/3] fix accelerate for testing --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index 2e87d75960ca..856a212e6022 100644 --- a/setup.py +++ b/setup.py @@ -177,6 +177,7 @@ def run(self): extras["docs"] = deps_list("hf-doc-builder") extras["training"] = deps_list("accelerate", "datasets", "tensorboard", "modelcards") extras["test"] = deps_list( + "accelerate", "datasets", "onnxruntime", "pytest", From 6fe1370ee6b70ddb70b5b94c21c53193a165fae0 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 5 Oct 2022 11:27:34 +0200 Subject: [PATCH 2/3] fix copies --- src/diffusers/utils/dummy_flax_objects.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/diffusers/utils/dummy_flax_objects.py b/src/diffusers/utils/dummy_flax_objects.py index 1e3ac002a609..4ab14f752c24 100644 --- a/src/diffusers/utils/dummy_flax_objects.py +++ b/src/diffusers/utils/dummy_flax_objects.py @@ -67,6 +67,13 @@ def __init__(self, *args, **kwargs): requires_backends(self, ["flax"]) +class FlaxSchedulerMixin(metaclass=DummyObject): + _backends = ["flax"] + + def __init__(self, *args, **kwargs): + requires_backends(self, ["flax"]) + + class FlaxScoreSdeVeScheduler(metaclass=DummyObject): _backends = ["flax"] From 94ebfff396c617400002dc46f37540eabaf24658 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 5 Oct 2022 11:31:17 +0200 Subject: [PATCH 3/3] uP --- setup.py | 1 - src/diffusers/dependency_versions_table.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 043bc7666994..007b4a2aa2e6 100644 --- a/setup.py +++ b/setup.py @@ -104,7 +104,6 @@ "torch>=1.4", "torchvision", "transformers>=4.21.0", - "accelerate>=0.12.0" ] # this is a lookup table with items like: diff --git a/src/diffusers/dependency_versions_table.py b/src/diffusers/dependency_versions_table.py index 5bde7340d2be..367b5c57a262 100644 --- a/src/diffusers/dependency_versions_table.py +++ b/src/diffusers/dependency_versions_table.py @@ -3,7 +3,7 @@ # 2. run `make deps_table_update`` deps = { "Pillow": "Pillow<10.0", - "accelerate": "accelerate>=0.12.0", + "accelerate": "accelerate>=0.11.0", "black": "black==22.8", "datasets": "datasets", "filelock": "filelock",