You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: docs/04-Finetune/02-Quick Start.md
+6-3Lines changed: 6 additions & 3 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -27,15 +27,16 @@ We recommend that you read the corresponding [model card](../05-Model%20Card.mdx
27
27
:::
28
28
29
29
1. Navigate to the `CogKit/` directory after cloning the repository
30
+
30
31
```bash
31
32
cd CogKit/
32
33
```
33
34
34
-
2. Choose the appropriate sub directory from the `quickstart/scripts` based on your task type and distribution strategy. For example, `t2i` corresponds to text-to-image task
35
+
2. Choose the appropriate subdirectory from the `quickstart/scripts` based on your task type and distribution strategy. For example, `t2i` corresponds to text-to-image task
35
36
36
37
3. Review and adjust the parameters in `config.yaml` in the selected training directory
37
38
38
-
4. Run the script in selected directory:
39
+
4. Run the script in the selected directory:
39
40
40
41
```bash
41
42
bash start_train.sh
@@ -45,7 +46,9 @@ We recommend that you read the corresponding [model card](../05-Model%20Card.mdx
45
46
46
47
### Merge Checkpoint
47
48
48
-
After fine-tuning, you need to use the `merge.py` script provided in the `quickstart/tools/converters` directory to merge the distributed checkpoint weights into a single checkpoint (**except for QLoRA fine-tuning**). For example:
49
+
After fine-tuning, you need to use the `merge.py` script to merge the distributed checkpoint weights into a single checkpoint (**except for QLoRA fine-tuning**).
50
+
The script can be found in the `quickstart/tools/converters` directory.
0 commit comments