From 90681989638b81930a3351b6b42484dc222236be Mon Sep 17 00:00:00 2001 From: yeungchenwa <523709335@qq.com> Date: Sat, 3 Feb 2024 18:01:49 +0800 Subject: [PATCH] [Update] --- README.md | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index be6549c..090eda5 100644 --- a/README.md +++ b/README.md @@ -51,9 +51,9 @@ - [x] Add phase 1 training and sampling script. - [x] Add WebUI demo. - [x] Push demo to Hugging Face. -- [ ] Combined with InstructPix2Pix. - [x] Add phase 2 training script and checkpoint. - [ ] Add the pre-training of SCR module. +- [ ] Combined with InstructPix2Pix. ## 🛠️ Installation ### Prerequisites (Recommended) @@ -114,11 +114,16 @@ The training data files tree should be (The data examples are shown in directory │ └── ... ``` ### Training Configuration -Before running the training script, you should set the training configuration, such as distributed training, through: +Before running the training script (including the following three modes), you should set the training configuration, such as distributed training, through: ```bash accelerate config ``` +### Training - Pretraining of SCR +```bash +Coming Soon ... +``` + ### Training - Phase 1 ```bash sh train_phase_1.sh