tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2025-12-06 04:12:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update Python version and torch dependencies

Browse Source 
Updated Python version requirement and adjusted torch dependencies for CPU, GPU, and AMD support.
This commit is contained in:
Lawrence R Kincheloe III 2025-11-22 20:02:59 -06:00 committed by GitHub
parent 1af926205d
commit dd37f29fe4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 48 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
pyproject.toml
View File

@ -3,7 +3,7 @@ name = "nanochat"
version = "0.1.0" version = "0.1.0"
description = "the minimal full-stack ChatGPT clone" description = "the minimal full-stack ChatGPT clone"
readme = "README.md" readme = "README.md"
requires-python = ">=3.10" requires-python = ">=3.12"
dependencies = [ dependencies = [
"datasets>=4.0.0", "datasets>=4.0.0",
"fastapi>=0.117.1", "fastapi>=0.117.1",
@ -13,7 +13,7 @@ dependencies = [
"setuptools>=80.9.0", "setuptools>=80.9.0",
"tiktoken>=0.11.0", "tiktoken>=0.11.0",
"tokenizers>=0.22.0", "tokenizers>=0.22.0",
"torch>=2.8.0", "torch>=2.7.0,<2.9.0",
"uvicorn>=0.36.0", "uvicorn>=0.36.0",
"wandb>=0.21.3", "wandb>=0.21.3",
] ]
@ -51,6 +51,20 @@ torch = [
{ index = "pytorch-rocm", extra = "amd" }, { index = "pytorch-rocm", extra = "amd" },
] ]
rocm = [
{ index = "pytorch-rocm", extra = "amd" },
]
pytorch-triton-rocm = [
{ index = "pytorch-rocm", extra = "amd" },
]
rocm-sdk-core = [
{ index = "pytorch-rocm", extra = "amd" },
]
rocm-sdk-libraries-gfx1151 = [
{ index = "pytorch-rocm", extra = "amd" },
]
[[tool.uv.index]] [[tool.uv.index]]
name = "pytorch-cpu" name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu" url = "https://download.pytorch.org/whl/cpu"
@ -63,21 +77,26 @@ explicit = true
[[tool.uv.index]] [[tool.uv.index]]
name = "pytorch-rocm" name = "pytorch-rocm"
url = "https://repo.amd.com/rocm/whl/" url = "https://repo.amd.com/rocm/whl/gfx1151"
explicit = true explicit = true
[project.optional-dependencies] [project.optional-dependencies]
cpu = [ cpu = [
"torch>=2.8.0", "torch>=2.7.0,<2.9.0",
] ]
gpu = [ gpu = [
"torch>=2.8.0", "torch>=2.7.0,<2.9.0",
] ]
amd = [ amd = [
"torch>=2.8.0", "torch>=2.7.0,<2.9.0",
"rocm",
"pytorch-triton-rocm",
"rocm-sdk-core",
"rocm-sdk-libraries-gfx1151",
] ]
[tool.uv] [tool.uv]
prerelease = "allow"
conflicts = [ conflicts = [
[ [
{ extra = "cpu" }, { extra = "cpu" },