File size: 281 Bytes
5e95049 | 1 2 3 4 5 6 7 8 9 10 11 | # Install a CUDA-matched torch FIRST, separately, e.g.: # pip install torch --index-url https://download.pytorch.org/whl/cu121 # (torch is intentionally not pinned here so you can match your GPU/CUDA.) lm_eval==0.4.12 sentencepiece tokenizers huggingface_hub datasets numpy tqdm |