| # ROCm port of requirements.txt for AMD gfx1151 (ROCm 7.13). |
| # Omitted on purpose — do not add them back: |
| # torch, torchvision: the target interpreter already has a working ROCm |
| # build (torch 2.10.0, torch.version.hip 7.13.99004). Reinstalling the |
| # upstream CUDA pins would replace it. |
| # transformer-engine: NVIDIA CUDA-only; ROCm has no TE. Qwen2RMSNorm in |
| # qwen2_5_vit.py is pure PyTorch, and the unused TE import is gone. |
| # Create the venv with system site packages so that ROCm torch is inherited: |
| # python3 -m venv --system-site-packages .venv |
| # .venv/bin/pip install -r requirements-rocm.txt |
| # |
| # Also omitted / relaxed versus upstream, because the ROCm interpreter is Python 3.13 |
| # and its torch is built against numpy 2.x: |
| # numpy upstream 1.23.1 has no Python 3.13 wheels, and downgrading would |
| # break the inherited torch. Inherit the system numpy (validated 2.2.4). |
| # Pillow upstream 10.4.0 has no Python 3.13 wheels. Inherit (validated 11.1.0). |
| # safetensors inherit the system build (validated 0.8.0). |
| # |
| # Validated on halo (gfx1151, ROCm 7.13, Python 3.13.5) on 2026-09-22: |
| # torch 2.10.0 (hip 7.13.99004) | numpy 2.2.4 | Pillow 11.1.0 | safetensors 0.8.0 |
| # transformers 4.57.1 | diffusers 0.36.0 | accelerate 1.13.0 | tokenizers 0.22.2 |
| # huggingface-hub 0.34.0 | peft 0.17.0 |
| transformers==4.57.1 |
| diffusers==0.36.0 |
| accelerate==1.13.0 |
| tokenizers==0.22.2 |
| huggingface-hub==0.34.0 |
| peft==0.17.0 |
| requests==2.32.3 |
| tqdm==4.67.1 |
| typing-extensions==4.15.0 |
|
|
| # Optional FlashAttention 2 backend (validated: flash-attn==2.7.3). The CLI |
| # default is eager attention; --attn-implementation flash_attention_2 needs |
| # this package. |
| # flash-attn==2.7.3 |
|
|
