Merge pull request #706 from svlandeg/fix/cpu

Add setuptools for CPU run
This commit is contained in:
Andrej
2026-04-14 11:33:14 -07:00
committed by GitHub
2 changed files with 3 additions and 0 deletions
+1
View File
@@ -54,6 +54,7 @@ explicit = true
[project.optional-dependencies] [project.optional-dependencies]
cpu = [ cpu = [
"setuptools>=65.0.0",
"torch==2.9.1", "torch==2.9.1",
] ]
gpu = [ gpu = [
Generated
+2
View File
@@ -1507,6 +1507,7 @@ dependencies = [
[package.optional-dependencies] [package.optional-dependencies]
cpu = [ cpu = [
{ name = "setuptools" },
{ name = "torch", version = "2.9.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform == 'darwin' and extra == 'extra-8-nanochat-cpu') or (extra == 'extra-8-nanochat-cpu' and extra == 'extra-8-nanochat-gpu')" }, { name = "torch", version = "2.9.1", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform == 'darwin' and extra == 'extra-8-nanochat-cpu') or (extra == 'extra-8-nanochat-cpu' and extra == 'extra-8-nanochat-gpu')" },
{ name = "torch", version = "2.9.1+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform != 'darwin' and extra == 'extra-8-nanochat-cpu') or (extra == 'extra-8-nanochat-cpu' and extra == 'extra-8-nanochat-gpu')" }, { name = "torch", version = "2.9.1+cpu", source = { registry = "https://download.pytorch.org/whl/cpu" }, marker = "(sys_platform != 'darwin' and extra == 'extra-8-nanochat-cpu') or (extra == 'extra-8-nanochat-cpu' and extra == 'extra-8-nanochat-gpu')" },
] ]
@@ -1530,6 +1531,7 @@ requires-dist = [
{ name = "kernels", specifier = ">=0.11.7" }, { name = "kernels", specifier = ">=0.11.7" },
{ name = "psutil", specifier = ">=7.1.0" }, { name = "psutil", specifier = ">=7.1.0" },
{ name = "rustbpe", specifier = ">=0.1.0" }, { name = "rustbpe", specifier = ">=0.1.0" },
{ name = "setuptools", marker = "extra == 'cpu'", specifier = ">=65.0.0" },
{ name = "tiktoken", specifier = ">=0.11.0" }, { name = "tiktoken", specifier = ">=0.11.0" },
{ name = "tokenizers", specifier = ">=0.22.0" }, { name = "tokenizers", specifier = ">=0.22.0" },
{ name = "torch", specifier = "==2.9.1" }, { name = "torch", specifier = "==2.9.1" },