feat: unify ControlNet model name option and deprecate old training script

This commit is contained in:
Kohya S
2024-12-07 17:28:19 +09:00
parent abff4b0ec7
commit e425996a59
2 changed files with 30 additions and 0 deletions

View File

@@ -15,6 +15,13 @@ The command to install PyTorch is as follows:
### Recent Updates
Dec 7, 2024:
- The option to specify the model name during ControlNet training was different in each script. It has been unified. Please specify `--controlnet_model_name_or_path`. PR [#1821](https://github.com/kohya-ss/sd-scripts/pull/1821) Thanks to sdbds!
<!--
Also, the ControlNet training script for SD has been changed from `train_controlnet.py` to `train_control_net.py`.
- `train_controlnet.py` is still available, but it will be removed in the future.
-->
- Fixed an issue where the saved model would be corrupted (pos_embed would not be saved) when `--enable_scaled_pos_embed` was specified in `sd3_train.py`.
Dec 3, 2024:

23
train_controlnet.py Normal file
View File

@@ -0,0 +1,23 @@
from library.utils import setup_logging
setup_logging()
import logging
logger = logging.getLogger(__name__)
from library import train_util
from train_control_net import setup_parser, train
if __name__ == "__main__":
logger.warning(
"The module 'train_controlnet.py' is deprecated. Please use 'train_control_net.py' instead"
" / 'train_controlnet.py'は非推奨です。代わりに'train_control_net.py'を使用してください。"
)
parser = setup_parser()
args = parser.parse_args()
train_util.verify_command_line_training_args(args)
args = train_util.read_config_from_file(args, parser)
train(args)