diff --git a/{{ cookiecutter.repo_name }}/pyproject.toml b/{{ cookiecutter.repo_name }}/pyproject.toml
index 3cdb2eb..515cbca 100644
--- a/{{ cookiecutter.repo_name }}/pyproject.toml	
+++ b/{{ cookiecutter.repo_name }}/pyproject.toml	
@@ -7,25 +7,20 @@ license = "MIT"
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    python = ">=3.10"
-    numpy = "^1.26.1"
-    pandas = "^2.1.1"
-    matplotlib = "^3.8.0"
-    scikit-learn = "^1.3.1"
-    loguru = "^0.7.2"
-    tqdm = "^4.66.1"
-    # einops = "^0.7.0"
-    # simple-parsing = "^0.1.4"
-    # 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"
-    # peft = "^0.5.0"
-    # loguru = "^0.7.2"
-    # einops = "^0.3.1"
+    "numpy=^1.26.1",
+    "pandas=^2.1.1",
+    "matplotlib=^3.8.0",
+    "scikit-learn=^1.3.1",
+    "loguru=^0.7.2",
+    "tqdm=^4.66.1",
+    "transformers[torch]=^4.34.0",
+    # "einops=^0.7.0",
+    # "loguru=^0.7.2",
+    # "simple-parsing=^0.1.4",
+    # "torch =^2.1.0+cu124",
+    # "lightning=^2.1.0",
+    # "torchinfo=^1.8.0",
+    # "peft=^0.5.0",
 ]
 
 [dependency-groups]