woops make xformers part of optional cuda group

pull/26/head
Guillermo Rodriguez 2023-10-03 13:37:44 -03:00
parent 537670d1f3
commit 08854562ef
No known key found for this signature in database
GPG Key ID: EC3AB66D5D83B392
1 changed files with 1 additions and 2 deletions

View File

@ -14,7 +14,6 @@ asks = '^3.0.0'
Pillow = '^10.0.1' Pillow = '^10.0.1'
docker = '^6.1.3' docker = '^6.1.3'
py-leap = {git = 'https://github.com/guilledk/py-leap.git', rev = 'v0.1a14'} py-leap = {git = 'https://github.com/guilledk/py-leap.git', rev = 'v0.1a14'}
xformers = "^0.0.22"
[tool.poetry.group.frontend] [tool.poetry.group.frontend]
optional = true optional = true
@ -42,6 +41,7 @@ scipy = {version = '^1.11.2'}
numba = {version = '0.57.0'} numba = {version = '0.57.0'}
triton = {version = '2.0.0', source = 'torch'} triton = {version = '2.0.0', source = 'torch'}
basicsr = {version = '^1.4.2'} basicsr = {version = '^1.4.2'}
xformers = {version = "^0.0.22"}
diffusers = {version = '^0.21.2'} diffusers = {version = '^0.21.2'}
realesrgan = {version = '^0.3.0'} realesrgan = {version = '^0.3.0'}
torchvision = {version = '0.15.2+cu118', source = 'torch'} torchvision = {version = '0.15.2+cu118', source = 'torch'}
@ -49,7 +49,6 @@ accelerate = {version = '^0.23.0'}
transformers = {version = '^4.33.2'} transformers = {version = '^4.33.2'}
huggingface-hub = {version = '^0.17.3'} huggingface-hub = {version = '^0.17.3'}
invisible-watermark = {version = '^0.2.0'} invisible-watermark = {version = '^0.2.0'}
# xformers = {git = 'https://github.com/facebookresearch/xformers.git', rev = 'v0.0.21'}
[[tool.poetry.source]] [[tool.poetry.source]]
name = 'torch' name = 'torch'