Skip to content

Commit be53b4e

Browse files
author
hanjian.thu123
committed
[update] add fine-tunning instructions
1 parent b67bf3c commit be53b4e

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

README.md

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,13 @@ bash scripts/eval.sh
156156
```
157157

158158
## Fine-tuning
159-
Fine-tuning Infinity is quite simple where you only need append ```--rush_resume=[infinity_vae_d32reg.pth]``` to [train.sh](scripts/train.sh).
159+
Fine-tuning Infinity is quite simple where you only need append ```--rush_resume=[infinity_vae_d32reg.pth]``` to [train.sh](scripts/train.sh). Note that you have to carefully set ```--pn``` for training and inference code since it decides the resolution of fine-tuning.
160+
161+
```
162+
--pn=0.06M # 256x256 resolution (including other aspect ratios with same number of pixels)
163+
--pn=0.25M # 512x512 resolution
164+
--pn=1M # 1024x1024 resolution
165+
```
160166

161167

162168
## One More Thing: Infinity-20B is coming soon 📆

0 commit comments

Comments
 (0)