- .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
|