mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-08 06:28:48 +00:00
feat: unify ControlNet model name option and deprecate old training script
This commit is contained in:
@@ -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
23
train_controlnet.py
Normal 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)
|
||||
Reference in New Issue
Block a user