diff --git a/{{ cookiecutter.repo_name }}/pyproject.toml b/{{ cookiecutter.repo_name }}/pyproject.toml
index 1e540be..d6e396a 100644
--- a/{{ cookiecutter.repo_name }}/pyproject.toml	
+++ b/{{ cookiecutter.repo_name }}/pyproject.toml	
@@ -8,18 +8,18 @@ readme = "README.md"
 
 [tool.poetry.dependencies]
 python = ">=3.10,<3.13"
-torch = {version = "^2.1.0+cu124", source = "pytorch"}
-simple-parsing = "^0.1.4"
+# simple-parsing = "^0.1.4"
 tqdm = "^4.66.1"
 numpy = "^1.26.1"
 pandas = "^2.1.1"
-lightning = "^2.1.0"
 matplotlib = "^3.8.0"
 loguru = "^0.7.2"
 einops = "^0.7.0"
 scikit-learn = "^1.3.1"
-pytorch-optimizer = "^2.12.0"
-torchinfo = "^1.8.0"
+# torch = {version = "^2.1.0+cu124", source = "pytorch"}
+# lightning = "^2.1.0"
+# pytorch-optimizer = "^2.12.0"
+# torchinfo = "^1.8.0"
 # accelerate = "^0.24.1"
 # transformers = "4.34.0"
 # accelerate = "^0.24.1"
@@ -27,17 +27,19 @@ torchinfo = "^1.8.0"
 # loguru = "^0.7.2"
 # einops = "^0.3.1"
 
+[tool.poetry.group.dev.dependencies]
+ipykernel = "^6.25.2"
+ruff = "^0.1.3"
+
 [[tool.poetry.source]]
 name = "pytorch"
 url = "https://download.pytorch.org/whl/cu124"
 priority = "explicit"
 
-
-[tool.poetry.group.dev.dependencies]
-ipykernel = "^6.25.2"
-ruff = "^0.1.3"
-pylama = "^8.4.1"
-
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
+
+[virtualenvs]
+create = true
+in-project = true