update readme

mlpc-ucsd · Sep 13, 2023 · b45425a · b45425a
1 parent e1be31c
commit b45425a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -89,23 +89,23 @@ After downloading the training datasets and specify their path in [dataset confi
 1. Pretraining of BLIVA's visual assistant branch 
 
 ```Shell
-torchrun --nnodes=1 --nproc_per_node=8\
+torchrun --nnodes=1 --nproc_per_node=8 \
     train.py \
     --cfg-path train_configs/pretrain_bliva_vicuna.yaml
 ```
 
 2. Instruction Finetuning BLIVA
 
 ```Shell
-torchrun --nnodes=1 --nproc_per_node=8\
+torchrun --nnodes=1 --nproc_per_node=8 \
     train.py \
     --cfg-path train_configs/finetune_bliva_vicuna.yaml
 ```
 
 Or, we also support training Vicuna7b together with BLIVA using LoRA during the second step, by default we don't use this version. 
 
 ```Shell
-torchrun --nnodes=1 --nproc_per_node=8\
+torchrun --nnodes=1 --nproc_per_node=8 \
     train.py \
     --cfg-path train_configs/finetune_bliva_and_vicuna.yaml
 ```