欢迎光临散文网 会员登陆 & 注册

【AI绘画】最佳人物模型训练!保姆式LoRA模型训练教程 一键包发布

2023-04-09 19:57 作者:不是一般的闷  | 我要投稿

各位大佬跪求帮忙!3070显卡 16G内存  --use_8bit_adam语段已经删除了,还是报错,求大佬指点一下.........

prepare tokenizer

update token length: 225

Use DreamBooth method.

prepare train images.

found directory 10_rentingyu contains 37 image files

370 train images with repeating.

loading image sizes.

100%|████████████████████████████████████████████████████████████████████████████████| 37/37 [00:00<00:00, 1686.33it/s]

make buckets

number of images (including repeats) / 各bucketの画像枚数(繰り返し回数を含む)

bucket 0: resolution (512, 512), count: 370

mean ar error (without repeats): 0.0

prepare accelerator

Using accelerator 0.15.0 or above.

load StableDiffusion checkpoint

Traceback (most recent call last):

  File "C:\Users\Administrator\AppData\Local\Programs\Python\Python310\lib\runpy.py", line 196, in _run_module_as_main

    return _run_code(code, main_globals, None,

  File "C:\Users\Administrator\AppData\Local\Programs\Python\Python310\lib\runpy.py", line 86, in _run_code

    exec(code, run_globals)

  File "F:\lora-scripts\venv\Scripts\accelerate.exe\__main__.py", line 7, in <module>

  File "F:\lora-scripts\venv\lib\site-packages\accelerate\commands\accelerate_cli.py", line 45, in main

    args.func(args)

  File "F:\lora-scripts\venv\lib\site-packages\accelerate\commands\launch.py", line 1104, in launch_command

    simple_launcher(args)

  File "F:\lora-scripts\venv\lib\site-packages\accelerate\commands\launch.py", line 567, in simple_launcher

    raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd)

subprocess.CalledProcessError: Command '['F:\\lora-scripts\\venv\\Scripts\\python.exe', './sd-scripts/train_network.py', '--enable_bucket', '--pretrained_model_name_or_path=./sd-models/chilloutmix_NiPrunedFp32Fix.safetensors', '--train_data_dir=./train/rentingyu', '--output_dir=./output', '--logging_dir=./logs', '--resolution=512,512', '--network_module=networks.lora', '--max_train_epochs=20', '--learning_rate=1e-4', '--unet_lr=1e-4', '--text_encoder_lr=1e-5', '--lr_scheduler=cosine_with_restarts', '--lr_warmup_steps=0', '--network_dim=32', '--network_alpha=32', '--output_name=rentyingyulora', '--train_batch_size=1', '--save_every_n_epochs=2', '--mixed_precision=fp16', '--save_precision=fp16', '--seed=1337', '--cache_latents', '--clip_skip=2', '--prior_loss_weight=1', '--max_token_length=225', '--caption_extension=.txt', '--save_model_as=safetensors', '--min_bucket_reso=256', '--max_bucket_reso=1024', '--xformers', '--shuffle_caption']' returned non-zero exit status 3221225477.

Train finished


【AI绘画】最佳人物模型训练!保姆式LoRA模型训练教程 一键包发布的评论 (共 条)

分享到微博请遵守国家法律