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"}