fix(readme): update cuda instructions during preprocess (#2114) [skip…

… ci]
axolotl-ai-cloud · Dec 3, 2024 · 4471a01 · 4471a01
1 parent 9f739da
commit 4471a01
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -147,7 +147,7 @@ pip3 install -e '.[flash-attn,deepspeed]'
 ### Usage
 ```bash
 # preprocess datasets - optional but recommended
-CUDA_VISIBLE_DEVICES="" python -m axolotl.cli.preprocess examples/openllama-3b/lora.yml
+CUDA_VISIBLE_DEVICES="0" python -m axolotl.cli.preprocess examples/openllama-3b/lora.yml
 
 # finetune lora
 accelerate launch -m axolotl.cli.train examples/openllama-3b/lora.yml