From 6e3e90db8696d5f8294a433b70268a1dca72aab5 Mon Sep 17 00:00:00 2001 From: Alex Date: Thu, 19 Mar 2026 10:23:02 -0700 Subject: clarify torch version --- README.md | 1 + requirements.txt | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index 958dff9..b684e58 100644 --- a/README.md +++ b/README.md @@ -126,6 +126,7 @@ By default, this command prints `train_loss` step logs during training and print For dataset export, tokenizer export, and docs-cache rebuild instructions, see [data/README.md](data/README.md). +Evaluation will be in the RunPod environment with all packages installed. `requirements.txt` is provided as a reference if you want to self-setup. ## FAQ diff --git a/requirements.txt b/requirements.txt index 0c5eedc..911b0e5 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ numpy tqdm -torch==2.10 +torch huggingface-hub kernels setuptools -- cgit v1.2.3