33 lines
624 B
TOML
33 lines
624 B
TOML
[project]
|
|
name = "attention-from-scratch"
|
|
version = "0.1.0"
|
|
description = "Add your description here"
|
|
readme = "README.md"
|
|
requires-python = ">=3.12"
|
|
dependencies = [
|
|
"ipykernel>=6.29.5",
|
|
"jupyter>=1.1.1",
|
|
"notebook>=7.4.1",
|
|
"numpy>=2.2.5",
|
|
"torch>=2.6.0",
|
|
]
|
|
|
|
|
|
[tool.uv.sources]
|
|
torch = [
|
|
{ index = "pytorch-cu124", marker = "sys_platform == 'linux'" },
|
|
]
|
|
torchvision = [
|
|
{ index = "pytorch-cu124", marker = "sys_platform == 'linux'" },
|
|
]
|
|
|
|
[[tool.uv.index]]
|
|
name = "pytorch-cu124"
|
|
url = "https://download.pytorch.org/whl/cu124"
|
|
explicit = true
|
|
|
|
[dependency-groups]
|
|
dev = [
|
|
"pre-commit>=4.2.0",
|
|
]
|