joey/SEC-cyBERT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

wrong pytorch somehow installed itself again lol

Browse Source
This commit is contained in:
Joey Eamigh 2026-03-30 20:15:16 -04:00
parent 3292980d33
commit 75ab92628b
No known key found for this signature in database
GPG Key ID: CE8C05DFFC53C9CB
1 changed files with 12 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
python/pyproject.toml
View File

@ -10,20 +10,26 @@ dependencies = [
"datasets",
"accelerate",
"pyyaml",
"nvidia-cusparselt-cu12>=0.8.1",
"nvidia-nvshmem-cu12>=3.6.5",
"nvidia-nvshmem-cu13>=3.4.5",
"nvidia-cuda-cccl>=13.2.27",
"flash-attn",
"unsloth",
]
[project.optional-dependencies]
flash = ["flash-attn"]
decoder = ["unsloth"]
[project.scripts]
sec-cybert = "main:main"
[[tool.uv.index]]
name = "pytorch-cu130"
url = "https://download.pytorch.org/whl/cu130"
explicit = true
[[tool.uv.index]]
url = "https://pypi.org/simple/"
default = true
[tool.uv.sources]
torch = [
{ index = "pytorch-cu130", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },
]
flash-attn = { url = "https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.9.4/flash_attn-2.6.3%2Bcu130torch2.11-cp313-cp313-linux_x86_64.whl" }