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