From 6595421894e5bb1bc3edb458f63d46ff8a67cb3c Mon Sep 17 00:00:00 2001 From: David Marx Date: Fri, 11 Nov 2022 09:37:15 -0800 Subject: [PATCH] added accelerate to hf deps --- pyproject.toml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 02dddc6..db948d1 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -29,8 +29,8 @@ dependencies = [ dynamic = ["version"] [project.optional-dependencies] -api = ["stability-sdk>=0.2.1"] -hf = ["diffusers","transformers","ftfy"] +api = ['stability-sdk>=0.2.1'] +hf = ['diffusers','transformers','ftfy','accelerate'] [tool.setuptools.packages.find] where =["."] @@ -38,4 +38,4 @@ include = ["vktrs*"] exclude = ["tests*"] [tool.setuptools.dynamic] -version = {file = "VERSION"} \ No newline at end of file +version = {file = "VERSION"}