Fix: Cannot tokenize with bf16 and on cpu (axolotl-ai-cloud#766)

mkeoliya · Oct 22, 2023 · 5da7a36 · 5da7a36
1 parent 1c1a59f
commit 5da7a36
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -819,7 +819,7 @@ accelerate launch -m axolotl.cli.train your_config.yml
 
 You can optionally pre-tokenize dataset with the following before finetuning:
 ```bash
-CUDA_VISIBLE_DEVICES="" accelerate launch -m axolotl.cli.train your_config.yml --prepare_ds_only
+CUDA_VISIBLE_DEVICES=0 accelerate launch -m axolotl.cli.train your_config.yml --prepare_ds_only
 ```
 
 ##### Config