hcpdiff 0.9.1__py3-none-any.whl → 2.2__py3-none-any.whl

hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples_noob/DreamBooth.yaml

@@ -1,70 +0,0 @@
-_base_:
-  - cfgs/train/dataset/regularization_dataset.yaml
-  - cfgs/train/train_base.yaml
-  - cfgs/train/tuning_base.yaml
-
-# 这部分根据需要改
-
-train:
-  save_step: 100
-
-model:
-  pretrained_model_name_or_path: 'runwayml/stable-diffusion-v1-5'
-  lr: 1e-6
-
-data_cfg:
-  batch_size: 4
-  img_root: 'imgs/' # 数据集路径
-  caption_file: 'imgs/' # 标注文件或文件夹
-  regularization_img_root: 'imgs/' # 正则化数据集
-  regularization_caption_file: 'imgs/' # 正则化数据集标注文件
-  target_area: 512*512 # 训练预期分辨率
-
-object_name: pt-cat1 # 训练物体名称
-class_name: cat # 训练类别名称
-
-
-
-# 这部分不用动
-
-unet:
-  -
-    lr: ${model.lr}
-    layers:
-      - ''
-
-data:
-  dataset1:
-    batch_size: ${data_cfg.batch_size}
-    cache_latents: True
-
-    source:
-      data_source1:
-        img_root: ${data_cfg.img_root}
-        prompt_template: 'prompt_tuning_template/object.txt'
-        caption_file: ${data_cfg.caption_file}
-
-        word_names:
-          pt1: ${object_name}
-          class: ${class_name}
-    bucket:
-      _target_: hcpdiff.data.bucket.RatioBucket.from_files # aspect ratio bucket
-      target_area: ${hcp.eval:"${data_cfg.target_area}"}
-      num_bucket: 1
-
-  dataset_class:
-    batch_size: 1
-    cache_latents: True
-    loss_weight: 1.0
-
-    source:
-      data_source1:
-        img_root: ${data_cfg.regularization_img_root}
-        prompt_template: 'prompt_tuning_template/object.txt'
-        caption_file: ${data_cfg.regularization_caption_file}
-
-        word_names:
-          class: ${class_name}
-    bucket:
-      _target_: hcpdiff.data.bucket.FixedBucket
-      target_size: [512, 512]

hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples_noob/TextualInversion.yaml

@@ -1,45 +0,0 @@
-_base_:
-  - cfgs/train/dataset/base_dataset.yaml
-  - cfgs/train/train_base.yaml
-  - cfgs/train/tuning_base.yaml
-
-# 这部分根据需要改
-
-train:
-  save_step: 100
-
-model:
-  pretrained_model_name_or_path: 'runwayml/stable-diffusion-v1-5'
-  lr: 1e-6
-  train_word_name: pt-catgirl1 # 训练单词名(需要提前创建单词)
-
-data_cfg:
-  batch_size: 4
-  img_root: 'imgs/' # 数据集路径
-  caption_file: 'imgs/' # 标注文件或文件夹
-  target_area: 512*512 # 训练预期分辨率
-
-
-
-
-
-# 这部分不用动
-
-data:
-  dataset1:
-    batch_size: ${data_cfg.batch_size}
-    cache_latents: True
-
-    source:
-      data_source1:
-        img_root: ${data_cfg.img_root}
-        prompt_template: 'prompt_tuning_template/object.txt'
-        caption_file: ${data_cfg.caption_file}
-
-        word_names:
-          pt1: ${model.train_word_name}
-
-    bucket:
-      _target_: hcpdiff.data.bucket.RatioBucket.from_files # aspect ratio bucket
-      target_area: ${hcp.eval:"${data_cfg.target_area}"}
-      num_bucket: 3

hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples_noob/fine-tuning.yaml

@@ -1,45 +0,0 @@
-_base_:
-  - cfgs/train/dataset/base_dataset.yaml
-  - cfgs/train/train_base.yaml
-  - cfgs/train/tuning_base.yaml
-
-# 这部分根据需要改
-
-train:
-  save_step: 100
-
-model:
-  pretrained_model_name_or_path: 'runwayml/stable-diffusion-v1-5'
-  lr: 1e-6
-
-data_cfg:
-  batch_size: 4
-  img_root: 'imgs/' # 数据集路径
-  caption_file: 'imgs/' # 标注文件或文件夹
-  target_area: 512*512 # 训练预期分辨率
-
-
-
-
-# 这部分不用动
-unet:
-  -
-    lr: ${model.lr}
-    layers:
-      - '' # fine-tuning all layers in unet
-
-data:
-  dataset1:
-    batch_size: ${data_cfg.batch_size}
-    cache_latents: True
-
-    source:
-      data_source1:
-        img_root: ${data_cfg.img_root}
-        prompt_template: 'prompt_tuning_template/object.txt'
-        caption_file: ${data_cfg.caption_file}
-
-    bucket:
-      _target_: hcpdiff.data.bucket.RatioBucket.from_files # aspect ratio bucket
-      target_area: ${hcp.eval:"${data_cfg.target_area}"}
-      num_bucket: 3

hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples_noob/lora.yaml

@@ -1,63 +0,0 @@
-_base_:
-  - cfgs/train/dataset/base_dataset.yaml
-  - cfgs/train/train_base.yaml
-  - cfgs/train/tuning_base.yaml
-
-
-# 这部分根据需要改
-
-train:
-  save_step: 100
-
-model:
-  pretrained_model_name_or_path: 'runwayml/stable-diffusion-v1-5'
-  lr_unet: 1e-4
-  lr_text_encoder: 1e-5
-  lora_rank: 8
-
-data_cfg:
-  batch_size: 4
-  img_root: 'imgs/' # 数据集路径
-  caption_file: 'imgs/' # 标注文件或文件夹
-  target_area: 512*512 # 训练预期分辨率
-
-object_name: pt-cat1 # lora训练的物体名(触发词)
-
-
-
-
-# 这部分不用动
-
-lora_unet:
-  -
-    lr: ${model.lr_unet}
-    rank: ${model.lora_rank}
-    layers:
-      - 're:.*\.attn.?$'
-      - 're:.*\.ff$'
-
-lora_text_encoder:
-  - lr: ${model.lr_text_encoder}
-    rank: ${model.lora_rank}
-    layers:
-      - 're:.*self_attn$'
-      - 're:.*mlp$'
-
-data:
-  dataset1:
-    batch_size: ${data_cfg.batch_size}
-    cache_latents: True
-
-    source:
-      data_source1:
-        img_root: ${data_cfg.img_root}
-        prompt_template: 'prompt_tuning_template/object.txt'
-        caption_file: ${data_cfg.caption_file}
-
-        word_names:
-          pt1: ${object_name}
-
-    bucket:
-      _target_: hcpdiff.data.bucket.RatioBucket.from_files # aspect ratio bucket
-      target_area: ${hcp.eval:"${data_cfg.target_area}"}
-      num_bucket: 3

hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/train_base.yaml

@@ -1,81 +0,0 @@
-
-exp_dir: exps/${hcp.time:}
-mixed_precision: 'fp16'
-allow_tf32: False
-seed: 114514
-ckpt_type: 'safetensors' # [torch, safetensors]
-
-vis_info:
-  prompt: null
-  negative_prompt: ''
-
-train:
-  train_steps: 1000
-  train_epochs: null # Choose one of [train_steps, train_epochs]
-  gradient_accumulation_steps: 1
-  workers: 4
-  max_grad_norm: 1.0
-  set_grads_to_none: False
-  save_step: 100
-  cfg_scale: '1.0' # for DreamArtist
-
-  resume: null
-#  resume:
-#    ckpt_path:
-#      unet: []
-#      TE: []
-#      words: {}
-#    start_step: 0
-
-  loss:
-    criterion:
-      _target_: torch.nn.MSELoss
-      _partial_: True
-      reduction: 'none' # support for attention mask
-    type: 'eps' # 'eps' or 'sample'
-
-  optimizer:
-    _target_: torch.optim.AdamW
-    _partial_: True
-    weight_decay: 1e-3
-
-  optimizer_pt:
-    _target_: torch.optim.AdamW
-    _partial_: True
-    weight_decay: 5e-4
-
-  scale_lr: True # auto scale lr with total batch size
-  scheduler:
-    name: 'one_cycle'
-    num_warmup_steps: 200
-    num_training_steps: 1000
-    scheduler_kwargs: {} # args for scheduler
-
-  scale_lr_pt: True
-  scheduler_pt: ${.scheduler}
-
-logger:
-  -
-    _target_: hcpdiff.loggers.CLILogger
-    _partial_: True
-    out_path: 'train.log'
-    log_step: 20
-
-model:
-  revision: null
-  pretrained_model_name_or_path: null
-  tokenizer_repeats: 2
-  enable_xformers: True
-  gradient_checkpointing: True
-  force_cast_precision: False
-  ema: null
-  clip_skip: 0
-  clip_final_norm: True
-
-  tokenizer: null
-  noise_scheduler: null
-  unet: null
-  text_encoder: null
-  vae: null
-
-previewer: null

hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/tuning_base.yaml

@@ -1,42 +0,0 @@
-# down部分总共6个 (Resnet+Transformer) 块，分别对应于:
-#   1:
-#     down_blocks.0.resnets.0
-#     down_blocks.0.attentions.0.transformer_blocks.0
-#   2:
-#     down_blocks.0.resnets.1
-#     down_blocks.0.attentions.1.transformer_blocks.0
-#   3:
-#     down_blocks.1.resnets.0
-#     down_blocks.1.attentions.0.transformer_blocks.0
-#   ......
-
-# up部分总共9个 (Resnet+Transformer) 块，分别对应于:
-#   1:
-#     up_blocks.1.resnets.0
-#     up_blocks.1.attentions.0.transformer_blocks.0
-#   2:
-#     up_blocks.1.resnets.1
-#     up_blocks.1.attentions.1.transformer_blocks.0
-#   3:
-#     up_blocks.1.resnets.2
-#     up_blocks.1.attentions.2.transformer_blocks.0
-#   4:
-#     up_blocks.2.resnets.0
-#     up_blocks.2.attentions.0.transformer_blocks.0
-#   ......
-
-tokenizer_pt:
-  emb_dir: 'embs/'
-  replace: False
-  train: []
-  #train:
-  #  - {name: pt1, lr: 1e-3}
-
-unet: null
-lora_unet: null
-
-text_encoder: null
-lora_text_encoder: null
-
-plugin_unet: null
-plugin_TE: null