Setup training dir

training/pyproject.toml

@@ -0,0 +1,66 @@
+[project]
+name = "NabuOCR"
+version = "0.1.0"
+description = "Neural Cuneiform Transliteration"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "unsloth>=2025.12.4",
+    "unsloth-zoo>=2025.12.3",
+    "triton-windows<3.6",
+    "xformers>=0.0.33",
+    "trl>=0.24.0",
+    "transformers>=4.45.1,<5.0.0",
+    "huggingface-hub[cli,hf-xet]>=0.34.4",
+    "ipython>=9.4.0",
+    "ipywidgets>=8.1.7",
+    "ipykernel>=6.30.1",
+    "holoviews>=1.21.0",
+    "prettytable>=3.16.0",
+    "nbformat>=5.10.4",
+    "qwen-vl-utils>=0.0.11",
+    "python-dotenv>=1.1.0",
+    "pillow>=12.0.0",
+    "einops>=0.8.1",
+    "jiwer>=4.0.0",
+    "addict>=2.4.0",
+    "easydict>=1.13",
+    "wcwidth>=0.2.14",
+    "pyxdameraulevenshtein>=1.9.0",
+]
+
+[project.optional-dependencies]
+cpu = ["torch>=2.9.0", "torchvision>=0.24.0", "torchaudio>=2.9.0"]
+cu128 = ["torch>=2.9.0", "torchvision>=0.24.0", "torchaudio>=2.9.0"]
+
+[tool.uv]
+environments = ["sys_platform == 'win32'"]
+conflicts = [[{ extra = "cpu" }, { extra = "cu128" }]]
+
+[tool.uv.sources]
+xformers = { index = "pytorch-cu128", marker = "platform_system == 'Windows'" }
+torch = [
+    { index = "pytorch-cpu", extra = "cpu" },
+    { index = "pytorch-cu128", extra = "cu128" },
+]
+torchaudio = [
+    { index = "pytorch-cpu", extra = "cpu" },
+    { index = "pytorch-cu128", extra = "cu128" },
+]
+torchvision = [
+    { index = "pytorch-cpu", extra = "cpu" },
+    { index = "pytorch-cu128", extra = "cu128" },
+]
+
+[[tool.uv.index]]
+name = "pytorch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+explicit = true
+
+[[tool.uv.index]]
+name = "pytorch-cu128"
+url = "https://download.pytorch.org/whl/cu128"
+explicit = true
+
+[tool.uv.extra-build-dependencies]
+flash-attn = ["torch"]