-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrequirements.txt
More file actions
24 lines (22 loc) · 1.46 KB
/
Copy pathrequirements.txt
File metadata and controls
24 lines (22 loc) · 1.46 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
# Radeon Cloud target: 64-bit Linux, CPython 3.12, ROCm 7.2.1.
# These AMD-published wheels provide the complete ROCm PyTorch runtime. They
# intentionally do not apply on Windows or non-CPython-3.12 environments.
# The host image must still provide the AMD driver and /dev/kfd device access.
torch @ https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.1/torch-2.9.1%2Brocm7.2.1.lw.gitff65f5bc-cp312-cp312-linux_x86_64.whl ; sys_platform == "linux" and python_version == "3.12" and platform_machine == "x86_64"
torchvision @ https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.1/torchvision-0.24.0%2Brocm7.2.1.gitb919bd0c-cp312-cp312-linux_x86_64.whl ; sys_platform == "linux" and python_version == "3.12" and platform_machine == "x86_64"
torchaudio @ https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.1/torchaudio-2.9.0%2Brocm7.2.1.gite3c6ee2b-cp312-cp312-linux_x86_64.whl ; sys_platform == "linux" and python_version == "3.12" and platform_machine == "x86_64"
triton @ https://repo.radeon.com/rocm/manylinux/rocm-rel-7.2.1/triton-3.5.1%2Brocm7.2.1.gita272dfa8-cp312-cp312-linux_x86_64.whl ; sys_platform == "linux" and python_version == "3.12" and platform_machine == "x86_64"
fastapi>=0.115,<1
uvicorn[standard]>=0.30,<1
pydantic>=2.10,<3
requests>=2.32,<3
tzdata>=2024.1; sys_platform == "win32"
# PixArt-Sigma inference stack.
diffusers>=0.30,<1
transformers>=4.45,<5
accelerate>=0.34,<2
safetensors>=0.4,<1
sentencepiece>=0.2,<1
ftfy>=6.2,<7
beautifulsoup4>=4.12,<5
pillow>=10,<12