AI绘画之训练自己的Lora模型

831

prepare tokenizer
update token length: 225
Use DreamBooth method.
prepare train images.
found directory 3_sucai contains 118 image files
354 train images with repeating.
loading image sizes.
100%|██████████████████████████████████████████████████████████████████████████████| 118/118 [00:00<00:00, 4534.47it/s]
make buckets
number of images (including repeats) / 各bucketの画像枚数（繰り返し回数を含む）
bucket 0: resolution (512, 640), count: 354
mean ar error (without repeats): 0.0
prepare accelerator
Using accelerator 0.15.0 or above.
load Diffusers pretrained models
Traceback (most recent call last):
  File "F:\lora-scripts\venv\lib\site-packages\diffusers\configuration_utils.py", line 326, in load_config
    config_file = hf_hub_download(
  File "F:\lora-scripts\venv\lib\site-packages\huggingface_hub\utils\_validators.py", line 114, in _inner_fn
    validate_repo_id(arg_value)
  File "F:\lora-scripts\venv\lib\site-packages\huggingface_hub\utils\_validators.py", line 166, in validate_repo_id
    raise HFValidationError(
huggingface_hub.utils._validators.HFValidationError: Repo id must be in the form 'repo_name' or 'namespace/repo_name': './sd-models/model.ckpt'. Use `repo_type` argument if needed.

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "F:\lora-scripts\sd-scripts\train_network.py", line 548, in <module>
    train(args)
  File "F:\lora-scripts\sd-scripts\train_network.py", line 156, in train
    text_encoder, vae, unet, _ = train_util.load_target_model(args, weight_dtype)
  File "F:\lora-scripts\sd-scripts\library\train_util.py", line 1587, in load_target_model
    pipe = StableDiffusionPipeline.from_pretrained(args.pretrained_model_name_or_path, tokenizer=None, safety_checker=None)
  File "F:\lora-scripts\venv\lib\site-packages\diffusers\pipeline_utils.py", line 454, in from_pretrained
    config_dict = cls.load_config(
  File "F:\lora-scripts\venv\lib\site-packages\diffusers\configuration_utils.py", line 363, in load_config
    raise EnvironmentError(
OSError: We couldn't connect to 'https://huggingface.co' to load this model, couldn't find it in the cached files and it looks like ./sd-models/model.ckpt is not the path to a directory containing a model_index.json file.
Checkout your internet connection or see how to run the library in offline mode at 'https://huggingface.co/docs/diffusers/installation#offline-mode'.
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/model.ckpt', '--train_data_dir=./train/sucai/', '--output_dir=./output', '--logging_dir=./logs', '--resolution=512,640', '--network_module=networks.lora', '--max_train_epochs=10', '--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=gaoye', '--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', '--use_8bit_adam']' returned non-zero exit status 1.
Train finished

抹茶味牛子

831 发表于 2023-2-21 08:38
prepare tokenizer
update token length: 225
Use DreamBooth method.

你加群吧，这个报错还是模型的问题，你是不是把SafeTensor格式的模型重命名成了ckpt

xueqing12

楼主你好 loss=nan怎么办呢？生成的模型毫无效果。

831

抹茶味牛子 发表于 2023-2-21 09:25
你加群吧，这个报错还是模型的问题，你是不是把SafeTensor格式的模型重命名成了ckpt ...

在群里问了，是模型出问题了，重新下载了模型后可以跑了，谢谢版主大人

longqisi

支出技术贴，刘明。

zuolian

我的电脑里为什么没有lora-scripts文件夹？明明我都在出图了 。。

ydybezd

学习学习

omesa

版主我运行训练ps文件最后就提示这个，怎么办呢


OSError: [WinError 1455] 页面文件太小，无法完成操作。 Error loading "F:\lora-scripts\lora-scripts\venv\lib\site-packages\torch\lib\cusolverMg64_11.dll" or one of its dependencies.

z10ym

omesa 发表于 2023-2-21 22:17
版主我运行训练ps文件最后就提示这个，怎么办呢

虚拟内存太小，根据路径，你需要增大F盘的虚拟内存

omesa

z10ym 发表于 2023-2-21 23:32
虚拟内存太小，根据路径，你需要增大F盘的虚拟内存

谢谢解答啊，加了一百多g虚拟内存已经完成了