mirror of
https://github.com/karpathy/nanochat.git
synced 2025-12-06 04:12:13 +00:00
Fix AMD ROCm install regression in speedrun.sh
Explicitly pass `--extra $EXTRAS` to `uv run` when building the tokenizer. This prevents `uv` from reverting to the default (NVIDIA) dependency set during the `maturin` build step, ensuring the correct PyTorch version (ROCm) is preserved on AMD hardware.
This commit is contained in:
parent
dd37f29fe4
commit
83bb650b49
|
|
@ -67,7 +67,7 @@ curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
|
|||
source "$HOME/.cargo/env"
|
||||
|
||||
# Build the rustbpe Tokenizer
|
||||
uv run maturin develop --release --manifest-path rustbpe/Cargo.toml
|
||||
uv run --extra $EXTRAS maturin develop --release --manifest-path rustbpe/Cargo.toml
|
||||
|
||||
# Download the first ~2B characters of pretraining dataset
|
||||
# look at dev/repackage_data_reference.py for details on how this data was prepared
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user