- .DS_Store +0 -0
- args.json +0 -99
- model_index.json +11 -2
- samples/.ipynb_checkpoints/0-checkpoint.png +0 -0
- samples/.ipynb_checkpoints/1-checkpoint.png +0 -0
- samples/.ipynb_checkpoints/2-checkpoint.png +0 -0
- samples/.ipynb_checkpoints/3-checkpoint.png +0 -0
- samples/0.png +0 -0
- samples/1.png +0 -0
- samples/2.png +0 -0
- samples/3.png +0 -0
- scheduler/scheduler_config.json +4 -2
- state.pt +0 -3
- {samples → unet}/.DS_Store +0 -0
- unet/config.json +11 -4
- unet/diffusion_pytorch_model.bin +2 -2
.DS_Store
CHANGED
|
Binary files a/.DS_Store and b/.DS_Store differ
|
|
|
args.json
DELETED
|
@@ -1,99 +0,0 @@
|
|
| 1 |
-
{
|
| 2 |
-
"pretrained_model_name_or_path": "./output/checkpoint_last",
|
| 3 |
-
"pretrained_vae_name_or_path": null,
|
| 4 |
-
"tokenizer_name": null,
|
| 5 |
-
"instance_data_dir": "./instance-images",
|
| 6 |
-
"class_data_dir": "./class-images",
|
| 7 |
-
"instance_prompt": "best quality, masterpiece, highres, Loliwm 1girl, Beautiful face",
|
| 8 |
-
"class_prompt": "best quality, masterpiece, highres, 1girl, Beautiful face",
|
| 9 |
-
"class_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",
|
| 10 |
-
"save_sample_prompt": "best quality, masterpiece, highres, Loliwm 1girl, Beautiful face, looking at viewer",
|
| 11 |
-
"save_sample_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",
|
| 12 |
-
"n_save_sample": 4,
|
| 13 |
-
"save_guidance_scale": 7.5,
|
| 14 |
-
"save_infer_steps": 50,
|
| 15 |
-
"with_prior_preservation": true,
|
| 16 |
-
"pad_tokens": false,
|
| 17 |
-
"prior_loss_weight": 1.0,
|
| 18 |
-
"num_class_images": 9295,
|
| 19 |
-
"output_dir": "./output",
|
| 20 |
-
"seed": 1337,
|
| 21 |
-
"resolution": 512,
|
| 22 |
-
"center_crop": false,
|
| 23 |
-
"train_text_encoder": false,
|
| 24 |
-
"train_batch_size": 1,
|
| 25 |
-
"sample_batch_size": 4,
|
| 26 |
-
"num_train_epochs": 1,
|
| 27 |
-
"max_train_steps": 3000,
|
| 28 |
-
"gradient_accumulation_steps": 1,
|
| 29 |
-
"gradient_checkpointing": true,
|
| 30 |
-
"learning_rate": 5e-06,
|
| 31 |
-
"scale_lr": false,
|
| 32 |
-
"scale_lr_sqrt": false,
|
| 33 |
-
"lr_scheduler": "cosine_with_restarts",
|
| 34 |
-
"lr_warmup_steps": 100,
|
| 35 |
-
"use_8bit_adam": true,
|
| 36 |
-
"use_deepspeed_adam": false,
|
| 37 |
-
"optimizer": "adamw",
|
| 38 |
-
"adam_beta1": 0.9,
|
| 39 |
-
"adam_beta2": 0.999,
|
| 40 |
-
"adam_epsilon": 1e-08,
|
| 41 |
-
"sgd_momentum": 0.9,
|
| 42 |
-
"sgd_dampening": 0,
|
| 43 |
-
"max_grad_norm": 1.0,
|
| 44 |
-
"weight_decay": 0.01,
|
| 45 |
-
"logging_dir": "logs",
|
| 46 |
-
"log_interval": 10,
|
| 47 |
-
"save_interval": 500,
|
| 48 |
-
"save_min_steps": 10,
|
| 49 |
-
"mixed_precision": "fp16",
|
| 50 |
-
"not_cache_latents": false,
|
| 51 |
-
"local_rank": -1,
|
| 52 |
-
"concepts_list": [
|
| 53 |
-
{
|
| 54 |
-
"instance_prompt": "best quality, masterpiece, highres, Loliwm 1girl, Beautiful face",
|
| 55 |
-
"class_prompt": "best quality, masterpiece, highres, 1girl, Beautiful face",
|
| 56 |
-
"class_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",
|
| 57 |
-
"instance_data_dir": "./instance-images",
|
| 58 |
-
"class_data_dir": "./class-images"
|
| 59 |
-
}
|
| 60 |
-
],
|
| 61 |
-
"wandb": false,
|
| 62 |
-
"wandb_artifact": false,
|
| 63 |
-
"rm_after_wandb_saved": false,
|
| 64 |
-
"wandb_name": "Stable-Diffusion-Dreambooth",
|
| 65 |
-
"read_prompt_filename": false,
|
| 66 |
-
"read_prompt_txt": false,
|
| 67 |
-
"append_prompt": "instance",
|
| 68 |
-
"save_unet_half": false,
|
| 69 |
-
"unet_half": false,
|
| 70 |
-
"clip_skip": 1,
|
| 71 |
-
"num_cycles": 1,
|
| 72 |
-
"last_epoch": -1,
|
| 73 |
-
"use_aspect_ratio_bucket": false,
|
| 74 |
-
"debug_arb": false,
|
| 75 |
-
"save_optimizer": true,
|
| 76 |
-
"save_states": true,
|
| 77 |
-
"resume": false,
|
| 78 |
-
"resume_from": "output/checkpoint_last",
|
| 79 |
-
"config": null,
|
| 80 |
-
"arb_dim_limit": 1024,
|
| 81 |
-
"arb_divisible": 64,
|
| 82 |
-
"arb_max_ar_error": 4,
|
| 83 |
-
"arb_max_size": [
|
| 84 |
-
768,
|
| 85 |
-
512
|
| 86 |
-
],
|
| 87 |
-
"arb_min_dim": 256,
|
| 88 |
-
"deepdanbooru": false,
|
| 89 |
-
"dd_threshold": 0.6,
|
| 90 |
-
"dd_alpha_sort": false,
|
| 91 |
-
"dd_use_spaces": true,
|
| 92 |
-
"dd_use_escape": true,
|
| 93 |
-
"enable_rotate": false,
|
| 94 |
-
"dd_include_ranks": false,
|
| 95 |
-
"use_ema": false,
|
| 96 |
-
"ucg": 0.0,
|
| 97 |
-
"debug_prompt": false,
|
| 98 |
-
"xformers": false
|
| 99 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
model_index.json
CHANGED
|
@@ -1,9 +1,18 @@
|
|
| 1 |
{
|
| 2 |
"_class_name": "StableDiffusionPipeline",
|
| 3 |
-
"_diffusers_version": "0.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
"scheduler": [
|
| 5 |
"diffusers",
|
| 6 |
-
"
|
| 7 |
],
|
| 8 |
"text_encoder": [
|
| 9 |
"transformers",
|
|
|
|
| 1 |
{
|
| 2 |
"_class_name": "StableDiffusionPipeline",
|
| 3 |
+
"_diffusers_version": "0.12.0.dev0",
|
| 4 |
+
"feature_extractor": [
|
| 5 |
+
"transformers",
|
| 6 |
+
"CLIPFeatureExtractor"
|
| 7 |
+
],
|
| 8 |
+
"requires_safety_checker": true,
|
| 9 |
+
"safety_checker": [
|
| 10 |
+
"stable_diffusion",
|
| 11 |
+
"StableDiffusionSafetyChecker"
|
| 12 |
+
],
|
| 13 |
"scheduler": [
|
| 14 |
"diffusers",
|
| 15 |
+
"PNDMScheduler"
|
| 16 |
],
|
| 17 |
"text_encoder": [
|
| 18 |
"transformers",
|
samples/.ipynb_checkpoints/0-checkpoint.png
DELETED
|
Binary file (355 kB)
|
|
|
samples/.ipynb_checkpoints/1-checkpoint.png
DELETED
|
Binary file (329 kB)
|
|
|
samples/.ipynb_checkpoints/2-checkpoint.png
DELETED
|
Binary file (277 kB)
|
|
|
samples/.ipynb_checkpoints/3-checkpoint.png
DELETED
|
Binary file (271 kB)
|
|
|
samples/0.png
DELETED
|
Binary file (355 kB)
|
|
|
samples/1.png
DELETED
|
Binary file (329 kB)
|
|
|
samples/2.png
DELETED
|
Binary file (277 kB)
|
|
|
samples/3.png
DELETED
|
Binary file (271 kB)
|
|
|
scheduler/scheduler_config.json
CHANGED
|
@@ -1,12 +1,14 @@
|
|
| 1 |
{
|
| 2 |
-
"_class_name": "
|
| 3 |
-
"_diffusers_version": "0.
|
| 4 |
"beta_end": 0.012,
|
| 5 |
"beta_schedule": "scaled_linear",
|
| 6 |
"beta_start": 0.00085,
|
| 7 |
"clip_sample": false,
|
| 8 |
"num_train_timesteps": 1000,
|
|
|
|
| 9 |
"set_alpha_to_one": false,
|
|
|
|
| 10 |
"steps_offset": 1,
|
| 11 |
"trained_betas": null
|
| 12 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"_class_name": "PNDMScheduler",
|
| 3 |
+
"_diffusers_version": "0.12.0.dev0",
|
| 4 |
"beta_end": 0.012,
|
| 5 |
"beta_schedule": "scaled_linear",
|
| 6 |
"beta_start": 0.00085,
|
| 7 |
"clip_sample": false,
|
| 8 |
"num_train_timesteps": 1000,
|
| 9 |
+
"prediction_type": "epsilon",
|
| 10 |
"set_alpha_to_one": false,
|
| 11 |
+
"skip_prk_steps": true,
|
| 12 |
"steps_offset": 1,
|
| 13 |
"trained_betas": null
|
| 14 |
}
|
state.pt
DELETED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:e05bd6b9d96c8d0e802eb9c1e374b0cdaa47d6c03dd2478b303f5894bfd9338e
|
| 3 |
-
size 1725045466
|
|
|
|
|
|
|
|
|
|
|
|
{samples → unet}/.DS_Store
RENAMED
|
Binary files a/samples/.DS_Store and b/unet/.DS_Store differ
|
|
|
unet/config.json
CHANGED
|
@@ -1,7 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"_class_name": "UNet2DConditionModel",
|
| 3 |
-
"_diffusers_version": "0.
|
| 4 |
-
"_name_or_path": "./output/checkpoint_last",
|
| 5 |
"act_fn": "silu",
|
| 6 |
"attention_head_dim": 8,
|
| 7 |
"block_out_channels": [
|
|
@@ -11,6 +10,7 @@
|
|
| 11 |
1280
|
| 12 |
],
|
| 13 |
"center_input_sample": false,
|
|
|
|
| 14 |
"cross_attention_dim": 768,
|
| 15 |
"down_block_types": [
|
| 16 |
"CrossAttnDownBlock2D",
|
|
@@ -19,19 +19,26 @@
|
|
| 19 |
"DownBlock2D"
|
| 20 |
],
|
| 21 |
"downsample_padding": 1,
|
|
|
|
| 22 |
"flip_sin_to_cos": true,
|
| 23 |
"freq_shift": 0,
|
| 24 |
"in_channels": 4,
|
| 25 |
"layers_per_block": 2,
|
| 26 |
"mid_block_scale_factor": 1,
|
|
|
|
| 27 |
"norm_eps": 1e-05,
|
| 28 |
"norm_num_groups": 32,
|
|
|
|
|
|
|
| 29 |
"out_channels": 4,
|
| 30 |
-
"
|
|
|
|
| 31 |
"up_block_types": [
|
| 32 |
"UpBlock2D",
|
| 33 |
"CrossAttnUpBlock2D",
|
| 34 |
"CrossAttnUpBlock2D",
|
| 35 |
"CrossAttnUpBlock2D"
|
| 36 |
-
]
|
|
|
|
|
|
|
| 37 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"_class_name": "UNet2DConditionModel",
|
| 3 |
+
"_diffusers_version": "0.12.0.dev0",
|
|
|
|
| 4 |
"act_fn": "silu",
|
| 5 |
"attention_head_dim": 8,
|
| 6 |
"block_out_channels": [
|
|
|
|
| 10 |
1280
|
| 11 |
],
|
| 12 |
"center_input_sample": false,
|
| 13 |
+
"class_embed_type": null,
|
| 14 |
"cross_attention_dim": 768,
|
| 15 |
"down_block_types": [
|
| 16 |
"CrossAttnDownBlock2D",
|
|
|
|
| 19 |
"DownBlock2D"
|
| 20 |
],
|
| 21 |
"downsample_padding": 1,
|
| 22 |
+
"dual_cross_attention": false,
|
| 23 |
"flip_sin_to_cos": true,
|
| 24 |
"freq_shift": 0,
|
| 25 |
"in_channels": 4,
|
| 26 |
"layers_per_block": 2,
|
| 27 |
"mid_block_scale_factor": 1,
|
| 28 |
+
"mid_block_type": "UNetMidBlock2DCrossAttn",
|
| 29 |
"norm_eps": 1e-05,
|
| 30 |
"norm_num_groups": 32,
|
| 31 |
+
"num_class_embeds": null,
|
| 32 |
+
"only_cross_attention": false,
|
| 33 |
"out_channels": 4,
|
| 34 |
+
"resnet_time_scale_shift": "default",
|
| 35 |
+
"sample_size": 64,
|
| 36 |
"up_block_types": [
|
| 37 |
"UpBlock2D",
|
| 38 |
"CrossAttnUpBlock2D",
|
| 39 |
"CrossAttnUpBlock2D",
|
| 40 |
"CrossAttnUpBlock2D"
|
| 41 |
+
],
|
| 42 |
+
"upcast_attention": false,
|
| 43 |
+
"use_linear_projection": false
|
| 44 |
}
|
unet/diffusion_pytorch_model.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:93323891d50baf144ead920cdab356d0767902f64b84913b11c5fb3b4381c2da
|
| 3 |
+
size 3438366373
|