Update README.md
Browse files
README.md
CHANGED
@@ -89,11 +89,11 @@ or check the inference notebook. before that, make sure you read the **Important
|
|
89 |
|
90 |
# Training:
|
91 |
|
92 |
-
First stage training
|
93 |
```bash
|
94 |
accelerate launch train_first.py --config_path ./Configs/config.yml
|
95 |
```
|
96 |
-
Second stage training
|
97 |
```bash
|
98 |
accelerate launch accelerate_train_second.py --config_path ./Configs/config.yml
|
99 |
```
|
@@ -106,8 +106,9 @@ or:
|
|
106 |
launch train_first.py --config_path ./Configs/config.yml
|
107 |
```
|
108 |
|
109 |
-
Third stage training (Kotodama, prompt encoding, etc.):
|
110 |
-
|
|
|
111 |
|
112 |
|
113 |
## some ideas for future
|
|
|
89 |
|
90 |
# Training:
|
91 |
|
92 |
+
**First stage training**:
|
93 |
```bash
|
94 |
accelerate launch train_first.py --config_path ./Configs/config.yml
|
95 |
```
|
96 |
+
**Second stage training**:
|
97 |
```bash
|
98 |
accelerate launch accelerate_train_second.py --config_path ./Configs/config.yml
|
99 |
```
|
|
|
106 |
launch train_first.py --config_path ./Configs/config.yml
|
107 |
```
|
108 |
|
109 |
+
**Third stage training** (Kotodama, prompt encoding, etc.):
|
110 |
+
```*maybe soon*
|
111 |
+
```
|
112 |
|
113 |
|
114 |
## some ideas for future
|