# 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