From 5d7c97b4deee3f7b21c14cf642e2e9fb76536572 Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Fri, 1 Dec 2023 22:58:29 +0800
Subject: [PATCH] update readme

Former-commit-id: 0cb260f453e480e5eb65b6424ac73e101b038ef1
---
 README.md    | 2 +-
 README_zh.md | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index bc20c1ce..c31196e9 100644
--- a/README.md
+++ b/README.md
@@ -250,7 +250,7 @@ export USE_MODELSCOPE_HUB=1 # `set USE_MODELSCOPE_HUB=1` for Windows
 Then you can train the corresponding model by specifying a model ID of the ModelScope Hub. (find a full list of model IDs at [ModelScope Hub](https://modelscope.cn/models))
 
 ```bash
-python src/train_bash.py \
+CUDA_VISIBLE_DEVICES=0 python src/train_bash.py \
     --model_name_or_path modelscope/Llama-2-7b-ms \
     ... # arguments (same as above)
 ```
diff --git a/README_zh.md b/README_zh.md
index 0aa8ac0f..6cc7a628 100644
--- a/README_zh.md
+++ b/README_zh.md
@@ -250,7 +250,7 @@ export USE_MODELSCOPE_HUB=1 # Windows 使用 `set USE_MODELSCOPE_HUB=1`
 接着即可通过指定模型名称来训练对应的模型。（在[魔搭社区](https://modelscope.cn/models)查看所有可用的模型）
 
 ```bash
-python src/train_bash.py \
+CUDA_VISIBLE_DEVICES=0 python src/train_bash.py \
     --model_name_or_path modelscope/Llama-2-7b-ms \
     ... # 参数同上
 ```