From ccde51c5ea6fc14dbbe627abb63c4631ad08cc9d Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Tue, 1 Aug 2023 18:48:27 +0800
Subject: [PATCH] update readme

---
 README.md    | 4 ++--
 README_zh.md | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index 4f0ff5b6..6674fe71 100644
--- a/README.md
+++ b/README.md
@@ -145,10 +145,10 @@ pip install https://github.com/jllllll/bitsandbytes-windows-webui/releases/downl
 ### All-in-one Web UI
 
 ```bash
-python src/train_web.py
+CUDA_VISIBLE_DEVICES=0 python src/train_web.py
 ```
 
-Currently the web UI only supports training on a single GPU.
+Currently the web UI only supports training on **a single GPU**.
 
 ### (Continually) Pre-Training
 
diff --git a/README_zh.md b/README_zh.md
index 88246178..125e278d 100644
--- a/README_zh.md
+++ b/README_zh.md
@@ -145,10 +145,10 @@ pip install https://github.com/jllllll/bitsandbytes-windows-webui/releases/downl
 ### 浏览器一键微调/测试
 
 ```bash
-python src/train_web.py
+CUDA_VISIBLE_DEVICES=0 python src/train_web.py
 ```
 
-目前网页 UI 仅支持单卡训练。
+目前网页 UI 仅支持**单卡训练**。
 
 ### 二次预训练