We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents fe529ee + e62af5d commit cbafb8fCopy full SHA for cbafb8f
1 file changed
docs/guides/checkpointing_solutions/convert_checkpoint.md
@@ -42,7 +42,7 @@ python3 -m pip install torch --index-url https://download.pytorch.org/whl/cpu
42
# Setup environment variables
43
export MODEL=<HF_MODEL> # e.g. 'llama3.1-8b-Instruct'
44
export BASE_OUTPUT_DIRECTORY=<CKPT_PATH> # e.g., gs://my-bucket/my-checkpoint-directory
45
-export USE_PATHWAYS=0 # Set to 1 for Pathways, 0 for McJAX
+export USE_PATHWAYS=0 # Set to 1 if you intend to use Pathways for training, 0 for McJAX
46
export LAZY_LOAD_TENSORS=<LAZY_LOAD> # Set to True to save RAM
47
```
48
0 commit comments