PyPI - hcpdiff - Versions diffs - 0.9.1__py3-none-any.whl → 2.2__py3-none-any.whl - Mend

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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (211) hide show

hcpdiff/__init__.py +4 -4
hcpdiff/ckpt_manager/__init__.py +4 -5
hcpdiff/ckpt_manager/ckpt.py +24 -0
hcpdiff/ckpt_manager/format/__init__.py +4 -0
hcpdiff/ckpt_manager/format/diffusers.py +59 -0
hcpdiff/ckpt_manager/format/emb.py +21 -0
hcpdiff/ckpt_manager/format/lora_webui.py +252 -0
hcpdiff/ckpt_manager/format/sd_single.py +41 -0
hcpdiff/ckpt_manager/loader.py +64 -0
hcpdiff/data/__init__.py +4 -28
hcpdiff/data/cache/__init__.py +1 -0
hcpdiff/data/cache/vae.py +102 -0
hcpdiff/data/dataset.py +20 -0
hcpdiff/data/handler/__init__.py +3 -0
hcpdiff/data/handler/controlnet.py +18 -0
hcpdiff/data/handler/diffusion.py +90 -0
hcpdiff/data/handler/text.py +111 -0
hcpdiff/data/source/__init__.py +3 -3
hcpdiff/data/source/folder_class.py +12 -29
hcpdiff/data/source/text.py +40 -0
hcpdiff/data/source/text2img.py +36 -74
hcpdiff/data/source/text2img_cond.py +9 -15
hcpdiff/diffusion/__init__.py +0 -0
hcpdiff/diffusion/noise/__init__.py +2 -0
hcpdiff/diffusion/noise/pyramid_noise.py +42 -0
hcpdiff/diffusion/noise/zero_terminal.py +39 -0
hcpdiff/diffusion/sampler/__init__.py +5 -0
hcpdiff/diffusion/sampler/base.py +72 -0
hcpdiff/diffusion/sampler/ddpm.py +20 -0
hcpdiff/diffusion/sampler/diffusers.py +66 -0
hcpdiff/diffusion/sampler/edm.py +22 -0
hcpdiff/diffusion/sampler/sigma_scheduler/__init__.py +3 -0
hcpdiff/diffusion/sampler/sigma_scheduler/base.py +14 -0
hcpdiff/diffusion/sampler/sigma_scheduler/ddpm.py +197 -0
hcpdiff/diffusion/sampler/sigma_scheduler/edm.py +48 -0
hcpdiff/easy/__init__.py +2 -0
hcpdiff/easy/cfg/__init__.py +3 -0
hcpdiff/easy/cfg/sd15_train.py +207 -0
hcpdiff/easy/cfg/sdxl_train.py +147 -0
hcpdiff/easy/cfg/t2i.py +228 -0
hcpdiff/easy/model/__init__.py +2 -0
hcpdiff/easy/model/cnet.py +31 -0
hcpdiff/easy/model/loader.py +79 -0
hcpdiff/easy/sampler.py +46 -0
hcpdiff/evaluate/__init__.py +1 -0
hcpdiff/evaluate/previewer.py +60 -0
hcpdiff/loss/__init__.py +4 -1
hcpdiff/loss/base.py +41 -0
hcpdiff/loss/gw.py +35 -0
hcpdiff/loss/ssim.py +37 -0
hcpdiff/loss/vlb.py +79 -0
hcpdiff/loss/weighting.py +66 -0
hcpdiff/models/__init__.py +2 -2
hcpdiff/models/cfg_context.py +17 -14
hcpdiff/models/compose/compose_hook.py +44 -23
hcpdiff/models/compose/compose_tokenizer.py +21 -8
hcpdiff/models/compose/sdxl_composer.py +4 -4
hcpdiff/models/controlnet.py +16 -16
hcpdiff/models/lora_base_patch.py +14 -25
hcpdiff/models/lora_layers.py +3 -9
hcpdiff/models/lora_layers_patch.py +14 -24
hcpdiff/models/text_emb_ex.py +84 -6
hcpdiff/models/textencoder_ex.py +54 -18
hcpdiff/models/wrapper/__init__.py +3 -0
hcpdiff/models/wrapper/pixart.py +19 -0
hcpdiff/models/wrapper/sd.py +218 -0
hcpdiff/models/wrapper/utils.py +20 -0
hcpdiff/parser/__init__.py +1 -0
hcpdiff/parser/embpt.py +32 -0
hcpdiff/tools/convert_caption_txt2json.py +1 -1
hcpdiff/tools/dataset_generator.py +94 -0
hcpdiff/tools/download_hf_model.py +24 -0
hcpdiff/tools/init_proj.py +3 -21
hcpdiff/tools/lora_convert.py +18 -17
hcpdiff/tools/save_model.py +12 -0
hcpdiff/tools/sd2diffusers.py +1 -1
hcpdiff/train_colo.py +1 -1
hcpdiff/train_deepspeed.py +1 -1
hcpdiff/trainer_ac.py +79 -0
hcpdiff/trainer_ac_single.py +31 -0
hcpdiff/utils/__init__.py +0 -2
hcpdiff/utils/inpaint_pipe.py +7 -2
hcpdiff/utils/net_utils.py +29 -6
hcpdiff/utils/pipe_hook.py +24 -7
hcpdiff/utils/utils.py +21 -4
hcpdiff/workflow/__init__.py +15 -10
hcpdiff/workflow/daam/__init__.py +1 -0
hcpdiff/workflow/daam/act.py +66 -0
hcpdiff/workflow/daam/hook.py +109 -0
hcpdiff/workflow/diffusion.py +118 -128
hcpdiff/workflow/fast.py +31 -0
hcpdiff/workflow/flow.py +67 -0
hcpdiff/workflow/io.py +36 -130
hcpdiff/workflow/model.py +46 -43
hcpdiff/workflow/text.py +60 -47
hcpdiff/workflow/utils.py +32 -12
hcpdiff/workflow/vae.py +37 -38
hcpdiff-2.2.dist-info/METADATA +299 -0
hcpdiff-2.2.dist-info/RECORD +115 -0
{hcpdiff-0.9.1.dist-info → hcpdiff-2.2.dist-info}/WHEEL +1 -1
hcpdiff-2.2.dist-info/entry_points.txt +5 -0
hcpdiff/ckpt_manager/base.py +0 -16
hcpdiff/ckpt_manager/ckpt_diffusers.py +0 -45
hcpdiff/ckpt_manager/ckpt_pkl.py +0 -138
hcpdiff/ckpt_manager/ckpt_safetensor.py +0 -64
hcpdiff/ckpt_manager/ckpt_webui.py +0 -54
hcpdiff/data/bucket.py +0 -358
hcpdiff/data/caption_loader.py +0 -80
hcpdiff/data/cond_dataset.py +0 -40
hcpdiff/data/crop_info_dataset.py +0 -40
hcpdiff/data/data_processor.py +0 -33
hcpdiff/data/pair_dataset.py +0 -146
hcpdiff/data/sampler.py +0 -54
hcpdiff/data/source/base.py +0 -30
hcpdiff/data/utils.py +0 -80
hcpdiff/deprecated/__init__.py +0 -1
hcpdiff/deprecated/cfg_converter.py +0 -81
hcpdiff/deprecated/lora_convert.py +0 -31
hcpdiff/infer_workflow.py +0 -57
hcpdiff/loggers/__init__.py +0 -13
hcpdiff/loggers/base_logger.py +0 -76
hcpdiff/loggers/cli_logger.py +0 -40
hcpdiff/loggers/preview/__init__.py +0 -1
hcpdiff/loggers/preview/image_previewer.py +0 -149
hcpdiff/loggers/tensorboard_logger.py +0 -30
hcpdiff/loggers/wandb_logger.py +0 -31
hcpdiff/loggers/webui_logger.py +0 -9
hcpdiff/loss/min_snr_loss.py +0 -52
hcpdiff/models/layers.py +0 -81
hcpdiff/models/plugin.py +0 -348
hcpdiff/models/wrapper.py +0 -75
hcpdiff/noise/__init__.py +0 -3
hcpdiff/noise/noise_base.py +0 -16
hcpdiff/noise/pyramid_noise.py +0 -50
hcpdiff/noise/zero_terminal.py +0 -44
hcpdiff/train_ac.py +0 -566
hcpdiff/train_ac_single.py +0 -39
hcpdiff/utils/caption_tools.py +0 -105
hcpdiff/utils/cfg_net_tools.py +0 -321
hcpdiff/utils/cfg_resolvers.py +0 -16
hcpdiff/utils/ema.py +0 -52
hcpdiff/utils/img_size_tool.py +0 -248
hcpdiff/vis/__init__.py +0 -3
hcpdiff/vis/base_interface.py +0 -12
hcpdiff/vis/disk_interface.py +0 -48
hcpdiff/vis/webui_interface.py +0 -17
hcpdiff/viser_fast.py +0 -138
hcpdiff/visualizer.py +0 -265
hcpdiff/visualizer_reloadable.py +0 -237
hcpdiff/workflow/base.py +0 -59
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/infer/anime/text2img_anime.yaml +0 -21
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/infer/anime/text2img_anime_lora.yaml +0 -58
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/infer/change_vae.yaml +0 -6
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/infer/euler_a.yaml +0 -8
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/infer/img2img.yaml +0 -10
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/infer/img2img_controlnet.yaml +0 -19
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/infer/inpaint.yaml +0 -11
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/infer/load_lora.yaml +0 -26
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/infer/load_unet_part.yaml +0 -18
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/infer/offload_2GB.yaml +0 -6
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/infer/save_model.yaml +0 -44
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/infer/text2img.yaml +0 -53
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/infer/text2img_DA++.yaml +0 -34
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/infer/text2img_sdxl.yaml +0 -9
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/plugins/plugin_controlnet.yaml +0 -17
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/te_struct.txt +0 -193
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/dataset/base_dataset.yaml +0 -29
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/dataset/regularization_dataset.yaml +0 -31
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/CustomDiffusion.yaml +0 -74
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/DreamArtist++.yaml +0 -135
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/DreamArtist.yaml +0 -45
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/DreamBooth.yaml +0 -62
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/FT_sdxl.yaml +0 -33
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/Lion_optimizer.yaml +0 -17
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/TextualInversion.yaml +0 -41
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/add_logger_tensorboard_wandb.yaml +0 -15
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/controlnet.yaml +0 -53
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/ema.yaml +0 -10
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/fine-tuning.yaml +0 -53
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/locon.yaml +0 -24
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/lora_anime_character.yaml +0 -77
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/lora_conventional.yaml +0 -56
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/lora_sdxl.yaml +0 -41
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/min_snr.yaml +0 -7
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples/preview_in_training.yaml +0 -6
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples_noob/DreamBooth.yaml +0 -70
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples_noob/TextualInversion.yaml +0 -45
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples_noob/fine-tuning.yaml +0 -45
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/examples_noob/lora.yaml +0 -63
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/train_base.yaml +0 -81
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/train/tuning_base.yaml +0 -42
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/unet_struct.txt +0 -932
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/workflow/highres_fix_latent.yaml +0 -86
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/workflow/highres_fix_pixel.yaml +0 -99
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/workflow/text2img.yaml +0 -59
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/workflow/text2img_lora.yaml +0 -70
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/zero2.json +0 -32
hcpdiff-0.9.1.data/data/hcpdiff/cfgs/zero3.json +0 -39
hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/caption.txt +0 -1
hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/name.txt +0 -1
hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/name_2pt_caption.txt +0 -1
hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/name_caption.txt +0 -1
hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/object.txt +0 -27
hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/object_caption.txt +0 -27
hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/style.txt +0 -19
hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/style_caption.txt +0 -19
hcpdiff-0.9.1.dist-info/METADATA +0 -199
hcpdiff-0.9.1.dist-info/RECORD +0 -160
hcpdiff-0.9.1.dist-info/entry_points.txt +0 -2
{hcpdiff-0.9.1.dist-info → hcpdiff-2.2.dist-info/licenses}/LICENSE +0 -0
{hcpdiff-0.9.1.dist-info → hcpdiff-2.2.dist-info}/top_level.txt +0 -0

hcpdiff-0.9.1.data/data/hcpdiff/cfgs/workflow/highres_fix_latent.yaml DELETED Viewed

@@ -1,86 +0,0 @@
-dtype: fp16
-bs: 2
-seed: 42
-memory: { }
-prepare:
-  - _target_: hcpdiff.workflow.LoadModelsAction
-    pretrained_model: 'ckpts/any5'
-    dtype: ${dtype}
-    scheduler:
-      _target_: diffusers.EulerAncestralDiscreteScheduler # change Sampler
-      beta_start: 0.00085
-      beta_end: 0.012
-      beta_schedule: 'scaled_linear'
-  - _target_: hcpdiff.workflow.XformersEnableAction
-  - _target_: hcpdiff.workflow.ExecAction
-    prog: |-
-      import torch
-      from hcpdiff.utils.net_utils import to_cpu, to_cuda
-      to_cuda(memory.unet)
-      to_cuda(memory.text_encoder)
-      memory.vae.to(dtype=torch.bfloat16)
-      #to_cuda(memory.vae)
-  - _target_: hcpdiff.workflow.PrepareDiffusionAction
-    dtype: ${dtype}
-  - _target_: hcpdiff.workflow.VaeOptimizeAction
-    slicing: True
-actions:
-  - _target_: hcpdiff.workflow.TextHookAction # text encoder and tokenizer auto get from memory
-    N_repeats: 1
-    layer_skip: 1
-  # encode text
-  - _target_: hcpdiff.workflow.AttnMultTextEncodeAction
-    prompt: 'masterpiece, best quality, 1girl, cat ears, outside'
-    negative_prompt: 'lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry'
-    bs: ${bs}
-  # prepare seed
-  - _target_: hcpdiff.workflow.SeedAction
-    seed: ${seed}
-  - _target_: hcpdiff.workflow.MakeTimestepsAction
-    N_steps: 30
-  # text to image
-  - _target_: hcpdiff.workflow.MakeLatentAction
-    width: 512
-    height: 768
-  - _target_: hcpdiff.workflow.LoopAction
-    loop_value:
-      timesteps: t
-    actions:
-      - _target_: hcpdiff.workflow.DiffusionStepAction
-        guidance_scale: 7.0
-  # image to image
-  - _target_: hcpdiff.workflow.LatentResizeAction
-  - _target_: hcpdiff.workflow.SeedAction
-    seed: ${seed}
-  - _target_: hcpdiff.workflow.MakeTimestepsAction
-    N_steps: 30
-  # only part of timesteps
-  - _target_: hcpdiff.workflow.ExecAction
-    prog: |-
-      states['timesteps'] = states['timesteps'][int(30*(1-0.6)):]
-      states['start_timestep'] = states['timesteps'][:1]
-  - _target_: hcpdiff.workflow.MakeLatentAction
-    width: 1024
-    height: 1536
-  - _target_: hcpdiff.workflow.LoopAction
-    loop_value:
-      timesteps: t
-    actions:
-      - _target_: hcpdiff.workflow.DiffusionStepAction
-        guidance_scale: 7.0
-  # decode to image
-  - _target_: hcpdiff.workflow.ExecAction
-    prog: |-
-      from hcpdiff.utils.net_utils import to_cpu, to_cuda
-      to_cpu(memory.unet)
-  - _target_: hcpdiff.workflow.DecodeAction
-    vae: ${hcp.from_memory:vae}
-    offload: true
-  - _target_: hcpdiff.workflow.SaveImageAction
-    save_root: output_pipe/
-    image_type: png

hcpdiff-0.9.1.data/data/hcpdiff/cfgs/workflow/highres_fix_pixel.yaml DELETED Viewed

@@ -1,99 +0,0 @@
-dtype: fp16
-bs: 2
-seed: 42
-memory: { }
-prepare:
-  - _target_: hcpdiff.workflow.LoadModelsAction
-    pretrained_model: 'ckpts/any5'
-    dtype: ${dtype}
-    scheduler:
-      _target_: diffusers.EulerAncestralDiscreteScheduler # change Sampler
-      beta_start: 0.00085
-      beta_end: 0.012
-      beta_schedule: 'scaled_linear'
-  - _target_: hcpdiff.workflow.XformersEnableAction
-  - _target_: hcpdiff.workflow.ExecAction
-    prog: |-
-      import torch
-      memory.vae.to(dtype=torch.bfloat16)
-  - _target_: hcpdiff.workflow.PrepareDiffusionAction
-    dtype: ${dtype}
-  - _target_: hcpdiff.workflow.VaeOptimizeAction
-    slicing: True
-actions:
-  - _target_: hcpdiff.workflow.TextHookAction # text encoder and tokenizer auto get from memory
-    N_repeats: 1
-    layer_skip: 1
-  # encode text
-  - _target_:  hcpdiff.workflow.StartTextEncode
-  - _target_: hcpdiff.workflow.AttnMultTextEncodeAction
-    prompt: 'masterpiece, best quality, 1girl, cat ears, outside'
-    negative_prompt: 'lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry'
-    bs: ${bs}
-  - _target_:  hcpdiff.workflow.EndTextEncode
-  - _target_:  hcpdiff.workflow.StartDiffusion
-  # prepare seed
-  - _target_: hcpdiff.workflow.SeedAction
-    seed: ${seed}
-  - _target_: hcpdiff.workflow.MakeTimestepsAction
-    N_steps: 30
-  # text to image
-  - _target_: hcpdiff.workflow.MakeLatentAction
-    width: 512
-    height: 768
-  - _target_: hcpdiff.workflow.LoopAction
-    loop_value:
-      timesteps: t
-    actions:
-      - _target_: hcpdiff.workflow.DiffusionStepAction
-        guidance_scale: 7.0
-  - _target_:  hcpdiff.workflow.EndDiffusion
-  # decode and resize image
-  - _target_: hcpdiff.workflow.DecodeAction
-    vae: ${hcp.from_memory:vae}
-    offload: false
-  - _target_: hcpdiff.workflow.SaveImageAction
-    save_root: output_pipe/
-    image_type: webp
-  - _target_: hcpdiff.workflow.ImageResizeAction
-  - _target_: hcpdiff.workflow.EncodeAction
-    vae: ${hcp.from_memory:vae}
-  # image to image
-  - _target_:  hcpdiff.workflow.StartDiffusion
-  - _target_: hcpdiff.workflow.SeedAction
-    seed: ${seed}
-  - _target_: hcpdiff.workflow.MakeTimestepsAction
-    N_steps: 30
-  # only part of timesteps
-  - _target_: hcpdiff.workflow.ExecAction
-    prog: |-
-      states['timesteps'] = states['timesteps'][int(30*(1-0.6)):]
-      states['start_timestep'] = states['timesteps'][:1]
-  - _target_: hcpdiff.workflow.MakeLatentAction
-    width: 1024
-    height: 1536
-  - _target_: hcpdiff.workflow.LoopAction
-    loop_value:
-      timesteps: t
-    actions:
-      - _target_: hcpdiff.workflow.DiffusionStepAction
-        guidance_scale: 7.0
-  - _target_:  hcpdiff.workflow.EndDiffusion
-  # decode to image
-  - _target_: hcpdiff.workflow.ExecAction
-    prog: |-
-      from hcpdiff.utils.net_utils import to_cpu, to_cuda
-      to_cpu(memory.unet)
-  - _target_: hcpdiff.workflow.DecodeAction
-    vae: ${hcp.from_memory:vae}
-    offload: true
-  - _target_: hcpdiff.workflow.SaveImageAction
-    save_root: output_pipe/
-    image_type: png

hcpdiff-0.9.1.data/data/hcpdiff/cfgs/workflow/text2img.yaml DELETED Viewed

@@ -1,59 +0,0 @@
-dtype: fp16
-amp: fp16
-bs: 2
-seed: 42
-memory: { }
-prepare:
-  - _target_: hcpdiff.workflow.LoadModelsAction
-    pretrained_model: 'ckpts/any5'
-    dtype: ${dtype}
-    scheduler:
-      _target_: diffusers.EulerAncestralDiscreteScheduler # change Sampler
-      beta_start: 0.00085
-      beta_end: 0.012
-      beta_schedule: 'scaled_linear'
-  - _target_: hcpdiff.workflow.XformersEnableAction
-  - _target_: hcpdiff.workflow.ExecAction
-    prog: |-
-      from hcpdiff.utils.net_utils import to_cpu, to_cuda
-      to_cuda(memory.unet)
-      to_cuda(memory.text_encoder)
-      to_cuda(memory.vae)
-  - _target_: hcpdiff.workflow.PrepareDiffusionAction
-    dtype: ${dtype}
-    amp: ${amp}
-  - _target_: hcpdiff.workflow.VaeOptimizeAction
-    slicing: True
-actions:
-  - _target_: hcpdiff.workflow.TextHookAction
-    N_repeats: 1
-    layer_skip: 1
-  # encode text
-  - _target_: hcpdiff.workflow.AttnMultTextEncodeAction
-    prompt: 'masterpiece, best quality, 1girl, cat ears, outside'
-    negative_prompt: 'lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry'
-    bs: ${bs}
-  # prepare seed
-  - _target_: hcpdiff.workflow.SeedAction
-    seed: ${seed}
-  - _target_: hcpdiff.workflow.MakeTimestepsAction
-    N_steps: 30
-  # text to image
-  - _target_: hcpdiff.workflow.MakeLatentAction
-    width: 512
-    height: 512
-  - _target_: hcpdiff.workflow.LoopAction
-    loop_value:
-      timesteps: t #迭代timesteps，每一步的存成t到states里
-    actions:
-      - _target_: hcpdiff.workflow.DiffusionStepAction
-        guidance_scale: 7.0
-  # decode to image
-  - _target_: hcpdiff.workflow.DecodeAction
-    vae: ${hcp.from_memory:vae}
-  - _target_: hcpdiff.workflow.SaveImageAction
-    save_root: output_pipe/
-    image_type: png

hcpdiff-0.9.1.data/data/hcpdiff/cfgs/workflow/text2img_lora.yaml DELETED Viewed

@@ -1,70 +0,0 @@
-dtype: fp16
-bs: 2
-seed: 42
-memory: { }
-prepare:
-  - _target_: hcpdiff.workflow.LoadModelsAction
-    pretrained_model: 'ckpts/any5'
-    dtype: ${dtype}
-    scheduler:
-      _target_: diffusers.EulerAncestralDiscreteScheduler # change Sampler
-      beta_start: 0.00085
-      beta_end: 0.012
-      beta_schedule: 'scaled_linear'
-  - _target_: hcpdiff.workflow.BuildModelLoaderAction
-  # load lora
-  - _target_: hcpdiff.workflow.LoadLoraAction
-    model: 'TE'
-    cfg:
-      - path: 'ckpts/nilu/text_encoder-1000.safetensors'
-        alpha: 0.8
-  - _target_: hcpdiff.workflow.LoadLoraAction
-    model: 'unet'
-    cfg:
-      - path: 'ckpts/nilu/unet-1000.safetensors'
-        alpha: 0.8
-  - _target_: hcpdiff.workflow.XformersEnableAction
-  - _target_: hcpdiff.workflow.ExecAction
-    prog: |-
-      from hcpdiff.utils.net_utils import to_cpu, to_cuda
-      to_cuda(memory.unet)
-      to_cuda(memory.vae)
-  - _target_: hcpdiff.workflow.PrepareDiffusionAction
-    dtype: ${dtype}
-  - _target_: hcpdiff.workflow.VaeOptimizeAction
-    slicing: True
-actions:
-  - _target_: hcpdiff.workflow.TextHookAction
-    N_repeats: 1
-    layer_skip: 1
-  # encode text
-  - _target_: hcpdiff.workflow.AttnMultTextEncodeAction
-    prompt: 'masterpiece, best quality, pt-xi1-1000, 1girl, cat ears, outside'
-    negative_prompt: 'lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry'
-    bs: ${bs}
-  # prepare seed
-  - _target_: hcpdiff.workflow.SeedAction
-    seed: ${seed}
-  - _target_: hcpdiff.workflow.MakeTimestepsAction
-    N_steps: 30
-  # text to image
-  - _target_: hcpdiff.workflow.MakeLatentAction
-    width: 512
-    height: 512
-  - _target_: hcpdiff.workflow.LoopAction
-    loop_value:
-      timesteps: t #迭代timesteps，每一步的存成t到states里
-    actions:
-      - _target_: hcpdiff.workflow.DiffusionStepAction
-        guidance_scale: 7.0
-  # decode to image
-  - _target_: hcpdiff.workflow.DecodeAction
-    vae: ${hcp.from_memory:vae}
-  - _target_: hcpdiff.workflow.SaveImageAction
-    save_root: output_pipe/
-    image_type: png

hcpdiff-0.9.1.data/data/hcpdiff/cfgs/zero2.json DELETED Viewed

@@ -1,32 +0,0 @@
-{
-    "fp16": {
-        "enabled": true,
-        "loss_scale": 0,
-        "loss_scale_window": 1000,
-        "initial_scale_power": 16,
-        "hysteresis": 2,
-        "min_loss_scale": 1
-    },
-    "bf16": {
-      "enabled": "auto"
-    },
-    "zero_optimization": {
-        "stage": 2,
-        "offload_optimizer": {
-            "device": "cpu",
-            "pin_memory": true
-        },
-        "allgather_partitions": true,
-        "allgather_bucket_size": 5e8,
-        "overlap_comm": true,
-        "reduce_scatter": true,
-        "reduce_bucket_size": 5e8,
-        "contiguous_gradients": true
-    },
-    "gradient_accumulation_steps": "auto",
-    "gradient_clipping": "auto",
-    "steps_per_print": 2000,
-    "train_batch_size": "auto",
-    "train_micro_batch_size_per_gpu": "auto",
-    "wall_clock_breakdown": false
-}

hcpdiff-0.9.1.data/data/hcpdiff/cfgs/zero3.json DELETED Viewed

@@ -1,39 +0,0 @@
-{
-    "fp16": {
-        "enabled": true,
-        "loss_scale": 0,
-        "loss_scale_window": 1000,
-        "initial_scale_power": 16,
-        "hysteresis": 2,
-        "min_loss_scale": 1
-    },
-    "bf16": {
-      "enabled": "auto"
-    },
-    "zero_optimization": {
-        "stage": 3,
-        "offload_optimizer": {
-            "device": "cpu",
-            "pin_memory": true
-        },
-        "offload_param": {
-            "device": "cpu",
-            "pin_memory": true
-        },
-        "overlap_comm": true,
-        "contiguous_gradients": true,
-        "reduce_bucket_size": 2e8,
-        "stage3_prefetch_bucket_size": 5e8,
-        "stage3_param_persistence_threshold": 1e6,
-        "sub_group_size": 1e9,
-        "stage3_max_live_parameters": 1e9,
-        "stage3_max_reuse_distance": 1e9,
-        "stage3_gather_16bit_weights_on_model_save": "auto"
-    },
-    "gradient_accumulation_steps": 1,
-    "gradient_clipping": "auto",
-    "steps_per_print": 2000,
-    "train_batch_size": "auto",
-    "train_micro_batch_size_per_gpu": "auto",
-    "wall_clock_breakdown": false
-}

hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/caption.txt DELETED Viewed

	@@ -1 +0,0 @@
1	- {pt1}{caption}

hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/name.txt DELETED Viewed

	@@ -1 +0,0 @@
1	- {name} {class}

hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/name_2pt_caption.txt DELETED Viewed

	@@ -1 +0,0 @@
1	- {pt1} {class1} with a {pt2} {class2} on hair, {caption}

hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/name_caption.txt DELETED Viewed

	@@ -1 +0,0 @@
1	- {name} {class}, {caption}

hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/object.txt DELETED Viewed

@@ -1,27 +0,0 @@
-a photo of a {pt1} {class}
-a rendering of a {pt1} {class}
-a cropped photo of the {pt1} {class}
-the photo of a {pt1} {class}
-a photo of a clean {pt1} {class}
-a photo of a dirty {pt1} {class}
-a dark photo of the {pt1} {class}
-a photo of my {pt1} {class}
-a photo of the cool {pt1} {class}
-a close-up photo of a {pt1} {class}
-a bright photo of the {pt1} {class}
-a cropped photo of a {pt1} {class}
-a photo of the {pt1} {class}
-a good photo of the {pt1} {class}
-a photo of one {pt1} {class}
-a close-up photo of the {pt1} {class}
-a rendition of the {pt1} {class}
-a photo of the clean {pt1} {class}
-a rendition of a {pt1} {class}
-a photo of a nice {pt1} {class}
-a good photo of a {pt1} {class}
-a photo of the nice {pt1} {class}
-a photo of the small {pt1} {class}
-a photo of the weird {pt1} {class}
-a photo of the large {pt1} {class}
-a photo of a cool {pt1} {class}
-a photo of a small {pt1} {class}

hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/object_caption.txt DELETED Viewed

@@ -1,27 +0,0 @@
-a photo of a {pt1} {class}, {caption}
-a rendering of a {pt1} {class}, {caption}
-a cropped photo of the {pt1} {class}, {caption}
-the photo of a {pt1} {class}, {caption}
-a photo of a clean {pt1} {class}, {caption}
-a photo of a dirty {pt1} {class}, {caption}
-a dark photo of the {pt1} {class}, {caption}
-a photo of my {pt1} {class}, {caption}
-a photo of the cool {pt1} {class}, {caption}
-a close-up photo of a {pt1} {class}, {caption}
-a bright photo of the {pt1} {class}, {caption}
-a cropped photo of a {pt1} {class}, {caption}
-a photo of the {pt1} {class}, {caption}
-a good photo of the {pt1} {class}, {caption}
-a photo of one {pt1} {class}, {caption}
-a close-up photo of the {pt1} {class}, {caption}
-a rendition of the {pt1} {class}, {caption}
-a photo of the clean {pt1} {class}, {caption}
-a rendition of a {pt1} {class}, {caption}
-a photo of a nice {pt1} {class}, {caption}
-a good photo of a {pt1} {class}, {caption}
-a photo of the nice {pt1} {class}, {caption}
-a photo of the small {pt1} {class}, {caption}
-a photo of the weird {pt1} {class}, {caption}
-a photo of the large {pt1} {class}, {caption}
-a photo of a cool {pt1} {class}, {caption}
-a photo of a small {pt1} {class}, {caption}

hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/style.txt DELETED Viewed

@@ -1,19 +0,0 @@
-a painting in the style of {pt1} {class}
-a rendering in the style of {pt1} {class}
-a cropped painting in the style of {pt1} {class}
-the painting in the style of {pt1} {class}
-a clean painting in the style of {pt1} {class}
-a dirty painting in the style of {pt1} {class}
-a dark painting in the style of {pt1} {class}
-a picture in the style of {pt1} {class}
-a cool painting in the style of {pt1} {class}
-a close-up painting in the style of {pt1} {class}
-a bright painting in the style of {pt1} {class}
-a cropped painting in the style of {pt1} {class}
-a good painting in the style of {pt1} {class}
-a close-up painting in the style of {pt1} {class}
-a rendition in the style of {pt1} {class}
-a nice painting in the style of {pt1} {class}
-a small painting in the style of {pt1} {class}
-a weird painting in the style of {pt1} {class}
-a large painting in the style of {pt1} {class}

hcpdiff-0.9.1.data/data/hcpdiff/prompt_tuning_template/style_caption.txt DELETED Viewed

@@ -1,19 +0,0 @@
-a painting in the style of {pt1} {class}, {caption}
-a rendering in the style of {pt1} {class}, {caption}
-a cropped painting in the style of {pt1} {class}, {caption}
-the painting in the style of {pt1} {class}, {caption}
-a clean painting in the style of {pt1} {class}, {caption}
-a dirty painting in the style of {pt1} {class}, {caption}
-a dark painting in the style of {pt1} {class}, {caption}
-a picture in the style of {pt1} {class}, {caption}
-a cool painting in the style of {pt1} {class}, {caption}
-a close-up painting in the style of {pt1} {class}, {caption}
-a bright painting in the style of {pt1} {class}, {caption}
-a cropped painting in the style of {pt1} {class}, {caption}
-a good painting in the style of {pt1} {class}, {caption}
-a close-up painting in the style of {pt1} {class}, {caption}
-a rendition in the style of {pt1} {class}, {caption}
-a nice painting in the style of {pt1} {class}, {caption}
-a small painting in the style of {pt1} {class}, {caption}
-a weird painting in the style of {pt1} {class}, {caption}
-a large painting in the style of {pt1} {class}, {caption}

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

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