Jeffiyyyy commited on
Commit
9842f67
1 Parent(s): 2650a4f
.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.8.0.dev0",
 
 
 
 
 
 
 
 
 
4
  "scheduler": [
5
  "diffusers",
6
- "DDIMScheduler"
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",
scheduler/scheduler_config.json CHANGED
@@ -1,12 +1,14 @@
1
  {
2
- "_class_name": "DDIMScheduler",
3
- "_diffusers_version": "0.8.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
  "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
 
 
 
 
unet/.DS_Store ADDED
Binary file (6.15 kB). View file
 
unet/config.json CHANGED
@@ -1,7 +1,6 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
- "_diffusers_version": "0.8.0.dev0",
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
- "sample_size": 32,
 
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:4e52df4f136fb50662f88db88a75b10bb68c88d759ba5067919497d5a09fd7d4
3
- size 3438375973
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:93323891d50baf144ead920cdab356d0767902f64b84913b11c5fb3b4381c2da
3
+ size 3438366373