Skip to content

Commit 9fc5952

Browse files
authored
Update README.md
1 parent 54e9cd2 commit 9fc5952

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

src/README.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -159,14 +159,14 @@ scripts/sd3/train_ddp_lora.sh configs/sd_task/sd35_lora_2order.yaml
159159
scripts/qwenimage/train_fsdp.sh configs/qwenimage_task/qwenimage_full.yaml
160160
```
161161

162-
> [!NOTE]Note on EMA
162+
> [!NOTE]
163163
> Due to memory limit, we did not add separate EMA for QwenImage, thus, we set `ema_decay_rate: 0` in the config. Users could switch FSDP2, which could fully support this.
164164
> https://github.com/inclusionAI/TwinFlow/blob/ab5808dfbbc7b3c5712d73652021e7318c93a39b/src/steerers/qwenimage/sft_fsdp.py#L132-L136
165165

166-
> [!NOTE]Note on LoRA Training
166+
> [!NOTE]
167167
> Switch to LoRA training is easy, we suggest you to refer to `src/steerers/stable_diffusion_3/sft_ddp_lora.py` to add LoRA, and set the method config like `configs/sd_task/sd35_lora_1order.yaml` or `configs/sd_task/sd35_lora_2order.yaml`
168168

169-
> [!NOTE]Note on QwenImage-Edit Training
169+
> [!NOTE]
170170
> Current modeling supports single reference image input for editing, you need to modify minor code to support edit training.
171171

172172
1. In the config, change `QwenImage` to `QwenImageEdit`.
@@ -196,4 +196,4 @@ loss = method.training_step(
196196
step=(global_step - 1),
197197
v=None
198198
)
199-
```
199+
```

0 commit comments

Comments
 (0)