Skip to content

Commit

Permalink
[Update]
Browse files Browse the repository at this point in the history
  • Loading branch information
yeungchenwa committed Feb 3, 2024
1 parent 58f9558 commit 9068198
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 9068198

Please sign in to comment.