Merge pull request #706 from svlandeg/fix/cpu
Add setuptools for CPU run
This commit is contained in:
@@ -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 = [
|
||||||
|
|||||||
@@ -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" },
|
||||||
|
|||||||
Reference in New Issue
Block a user