to uv

pyproject.toml

@@ -1,34 +1,30 @@
-[tool.poetry]
+[project]
 name = "simpo"
 version = "0.1.0"
 description = ""
-authors = ["wassname <1103714+wassname@users.noreply.github.com>"]
 readme = "README.md"
-package-mode = false
+requires-python = ">=3.10.9,<4.0"
+
+dependencies = [
+    "torch>=2.1.2",
+    "deepspeed>=0.14",
+    "wandb>=0.18.2",
+    # git‐based dep
+    "alignment-handbook",
+]
+
+[dependency-groups]
+dev = [
+    "pytest",
+    "ipykernel>=6.29.5",
+]
 
 [build-system]
-requires = ["poetry-core"]
-build-backend = "poetry.core.masonry.api"
+requires = ["hatchling"]
+build-backend = "hatchling.build"
 
-[[tool.poetry.source]]
-name = "pytorch"
-url = "https://download.pytorch.org/whl/cu121"
-priority = "explicit"
+[tool.hatch.build.targets.wheel]
+packages = ["simpo"]
 
-[tool.poetry.dependencies]
-python = "^3.10.9,<4.0"
-torch = { version = ">=2.1.2+cu121", source = "pytorch" }
-alignment-handbook = {git = "https://github.com/huggingface/alignment-handbook.git"}
-#huggingface-hub = "~0.23.0"
-# trl = "~0.9.0"
-deepspeed = "^0.14"
-# transformers="~4.39.3"
-# python -m pip install flash-attn --no-build-isolation
-wandb = "^0.18.2"
-
-[tool.poetry.group.dev.dependencies]
-pytest = "*"
-ipykernel = "^6.29.5"
-
-[virtualenvs]
-in-project = true
+[tool.uv.sources]
+alignment-handbook = { git = "https://github.com/huggingface/alignment-handbook.git" }