Skip to content

Commit

Permalink
docs: update README with FLUX.1 ControlNet training details and impro…
Browse files Browse the repository at this point in the history
…ve argument help text
  • Loading branch information
kohya-ss committed Dec 2, 2024
1 parent 09a3740 commit e369b9a
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 2 deletions.
10 changes: 9 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,15 @@ The command to install PyTorch is as follows:

### Recent Updates

1 Dec, 2024:
Dec 2, 2024:

- FLUX.1 ControlNet training is supported. PR [#1813](https://github.com/kohya-ss/sd-scripts/pull/1813). Thanks to minux302! See PR and [here](#flux1-controlnet-training) for details.
- Not fully tested. Feedback is welcome.
- 80GB VRAM is required for 1024x1024 resolution, and 48GB VRAM is required for 512x512 resolution.
- Currently, it only works in Linux environment (or Windows WSL2) because DeepSpeed is required.
- Multi-GPU training is not tested.

Dec 1, 2024:

- Pseudo Huber loss is now available for FLUX.1 and SD3.5 training. See PR [#1808](https://github.com/kohya-ss/sd-scripts/pull/1808) for details. Thanks to recris!
- Specify `--loss_type huber` or `--loss_type smooth_l1` to use it. `--huber_c` and `--huber_scale` are also available.
Expand Down
2 changes: 1 addition & 1 deletion library/flux_train_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -567,7 +567,7 @@ def add_flux_train_arguments(parser: argparse.ArgumentParser):
"--controlnet",
type=str,
default=None,
help="path to controlnet (*.sft or *.safetensors) / aeのパス(*.sftまたは*.safetensors)"
help="path to controlnet (*.sft or *.safetensors) / controlnetのパス(*.sftまたは*.safetensors)"
)
parser.add_argument(
"--t5xxl_max_token_length",
Expand Down

0 comments on commit e369b9a

Please sign in to comment.