85 lines
2.7 KiB
Plaintext
85 lines
2.7 KiB
Plaintext
# NVIDIA RTX Spark — native Windows-on-ARM (win_arm64) requirement set.
|
|
#
|
|
# Mirrors requirements_base.txt with pins adjusted to versions that actually
|
|
# publish win_arm64 wheels (or that we build ourselves — see manager/spec.py
|
|
# spark handling). Keep in sync with requirements_base.txt when bumping pins.
|
|
#
|
|
# Deliberate differences from requirements_base.txt:
|
|
# - scipy>=1.17 first release with win_arm64 wheels (also unlocks numpy 2)
|
|
# - matplotlib>=3.11 first release with win_arm64 wheels (base: 3.10.1)
|
|
# - av>=17 first release with win_arm64 wheels (base: 16.0.1)
|
|
# - numba/llvmlite rc only prerelease win_arm64 wheels exist (librosa dep)
|
|
# - tensorboard REMOVED: its grpcio dep has no win_arm64 wheels; the
|
|
# manager installs tensorboard --no-deps (writer path
|
|
# only needs protobuf) as a spark extra
|
|
# - hf_transfer REMOVED: no win_arm64 wheels; hf-xet (arm64 OK) is the
|
|
# modern replacement and installs via manager extras
|
|
# - torchcodec pin REMOVED: the manager installs the locally built
|
|
# win_arm64 torchcodec wheel as an extra
|
|
# - opencv-python, soxr, pywavelets, brotli(gradio), kornia-rs have no
|
|
# published win_arm64 wheels at any version — supplied from the ai-toolkit
|
|
# spark wheel set (self-built), resolved via --find-links
|
|
|
|
numpy>=2,<3
|
|
scipy>=1.17
|
|
|
|
torchao==0.17.0
|
|
safetensors
|
|
git+https://github.com/huggingface/diffusers.git@c943837899b16cbae2f619b8dd4f7bb6f07dd81a
|
|
transformers==5.5.3
|
|
lycoris-lora==1.8.3
|
|
flatten_json
|
|
pyyaml
|
|
oyaml
|
|
kornia
|
|
invisible-watermark
|
|
einops
|
|
accelerate
|
|
toml
|
|
albumentations==1.4.15
|
|
albucore==0.0.16
|
|
pydantic
|
|
omegaconf
|
|
open_clip_torch
|
|
timm==1.0.22
|
|
prodigyopt
|
|
controlnet_aux==0.0.10
|
|
python-dotenv
|
|
bitsandbytes
|
|
lpips
|
|
pytorch_fid
|
|
optimum-quanto==0.2.4
|
|
sentencepiece
|
|
huggingface_hub==1.23.0
|
|
peft==0.18.1
|
|
gradio
|
|
python-slugify
|
|
# exact pins: newer opencv versions exist on PyPI only as sdists (which fail
|
|
# to build on MSVC arm64 — dnn __fp16); force the self-built 4.12 wheels
|
|
opencv-python==4.12.0.88
|
|
opencv-python-headless==4.12.0.88
|
|
pytorch-wavelets==1.3.0
|
|
matplotlib>=3.11
|
|
setuptools>=77.0.3
|
|
av>=17
|
|
# librosa chain: numba/llvmlite have no cp312 win_arm64 wheels upstream — the
|
|
# rc pins below resolve from the self-built spark wheel set (llvmlite built
|
|
# against our own LLVM 22 static build)
|
|
librosa==0.11.0
|
|
numba==0.67.0rc1
|
|
llvmlite==0.49.0rc1
|
|
soxr==1.1.0
|
|
mutagen==1.47.0
|
|
soundfile
|
|
|
|
# tensorboard is installed --no-deps by the manager (its grpcio dep has no
|
|
# win_arm64 wheels; grpc is only needed by the tensorboard server, not the
|
|
# SummaryWriter path the toolkit uses). Its remaining runtime deps:
|
|
absl-py
|
|
markdown
|
|
werkzeug
|
|
tensorboard-data-server
|
|
packaging
|
|
protobuf
|
|
six
|