【AI繪畫】最佳人物模型訓練!保姆式LoRA模型訓練教程 一鍵包發(fā)布
各位大佬跪求幫忙!3070顯卡 16G內(nèi)存??--use_8bit_adam語段已經(jīng)刪除了,還是報錯,求大佬指點一下.........
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の畫像枚數(shù)(繰り返し回數(shù)を含む)
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