File size: 367 Bytes
212a146 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | # Use the PyTorch ROCm wheel index (choose the stream that matches your system) --extra-index-url https://download.pytorch.org/whl/rocm6.3 pre-commit torch==2.8.0+rocm6.3 torchaudio==2.8.0+rocm6.3 torchvision==0.23.0+rocm6.3 transformers==4.57.1 qwen-vl-utils==0.0.11 datasets setuptools tqdm wandb psutil numpy accelerate pydantic sglang[all]==0.5.4 openai-harmony |