From d02bfe7879043c9a5ad6f00d8907664c1b740157 Mon Sep 17 00:00:00 2001
From: wangshuai6 <wangshuai.moonlight@bytedance.com>
Date: Mon, 14 Apr 2025 09:01:22 +0800
Subject: [PATCH] fix bugs(update README)

---
 README.md | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/README.md b/README.md
index 7592f34..d474576 100644
--- a/README.md
+++ b/README.md
@@ -101,6 +101,8 @@ data:
 ```bash
 # for inference
 python main.py predict -c configs/repa_improved_ddt_xlen22de6_256.yaml --ckpt_path=XXX.ckpt
+# # or specify the GPU(s) to use with as :
+CUDA_VISIBLE_DEVICES=0,1, python main.py predict -c configs/repa_improved_ddt_xlen22de6_256.yaml --ckpt_path=XXX.ckpt
 ```
 
 ```bash