mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-15 16:39:42 +00:00
16 lines
415 B
Python
16 lines
415 B
Python
import train_leco
|
|
from library import deepspeed_utils, train_util
|
|
|
|
|
|
def test_syntax():
|
|
assert train_leco is not None
|
|
|
|
|
|
def test_setup_parser_supports_shared_training_validation():
|
|
args = train_leco.setup_parser().parse_args(["--prompts_file", "slider.yaml"])
|
|
|
|
train_util.verify_training_args(args)
|
|
|
|
assert args.min_snr_gamma is None
|
|
assert deepspeed_utils.prepare_deepspeed_plugin(args) is None
|