From afedc470bd77bbd3713d480ac9d5e3d3625d522a Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Mon, 23 Oct 2023 01:32:26 +0900 Subject: [PATCH] Fix: Cannot tokenize with bf16 and on cpu (#766) --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index aec804da3..0512dee48 100644 --- 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