milsmo commited on
Commit
5e93cb1
·
verified ·
1 Parent(s): 4fe308e

End of training

Browse files
model_index.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "StableDiffusionPipeline",
3
  "_diffusers_version": "0.31.0",
4
- "_name_or_path": "runwayml/stable-diffusion-inpainting",
5
  "feature_extractor": [
6
  "transformers",
7
  "CLIPImageProcessor"
@@ -10,14 +10,14 @@
10
  null,
11
  null
12
  ],
13
- "requires_safety_checker": true,
14
  "safety_checker": [
15
- "stable_diffusion",
16
- "StableDiffusionSafetyChecker"
17
  ],
18
  "scheduler": [
19
  "diffusers",
20
- "DDIMScheduler"
21
  ],
22
  "text_encoder": [
23
  "transformers",
 
1
  {
2
  "_class_name": "StableDiffusionPipeline",
3
  "_diffusers_version": "0.31.0",
4
+ "_name_or_path": "stabilityai/stable-diffusion-2-inpainting",
5
  "feature_extractor": [
6
  "transformers",
7
  "CLIPImageProcessor"
 
10
  null,
11
  null
12
  ],
13
+ "requires_safety_checker": false,
14
  "safety_checker": [
15
+ null,
16
+ null
17
  ],
18
  "scheduler": [
19
  "diffusers",
20
+ "PNDMScheduler"
21
  ],
22
  "text_encoder": [
23
  "transformers",
scheduler/scheduler_config.json CHANGED
@@ -1,20 +1,15 @@
1
  {
2
- "_class_name": "DDIMScheduler",
3
  "_diffusers_version": "0.31.0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
  "clip_sample": false,
8
- "clip_sample_range": 1.0,
9
- "dynamic_thresholding_ratio": 0.995,
10
  "num_train_timesteps": 1000,
11
  "prediction_type": "epsilon",
12
- "rescale_betas_zero_snr": false,
13
- "sample_max_value": 1.0,
14
  "set_alpha_to_one": false,
15
  "skip_prk_steps": true,
16
  "steps_offset": 1,
17
- "thresholding": false,
18
  "timestep_spacing": "leading",
19
  "trained_betas": null
20
  }
 
1
  {
2
+ "_class_name": "PNDMScheduler",
3
  "_diffusers_version": "0.31.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
  "prediction_type": "epsilon",
 
 
10
  "set_alpha_to_one": false,
11
  "skip_prk_steps": true,
12
  "steps_offset": 1,
 
13
  "timestep_spacing": "leading",
14
  "trained_betas": null
15
  }
text_encoder/config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "runwayml/stable-diffusion-inpainting",
3
  "architectures": [
4
  "CLIPTextModel"
5
  ],
@@ -7,18 +7,18 @@
7
  "bos_token_id": 0,
8
  "dropout": 0.0,
9
  "eos_token_id": 2,
10
- "hidden_act": "quick_gelu",
11
- "hidden_size": 768,
12
  "initializer_factor": 1.0,
13
  "initializer_range": 0.02,
14
- "intermediate_size": 3072,
15
  "layer_norm_eps": 1e-05,
16
  "max_position_embeddings": 77,
17
  "model_type": "clip_text_model",
18
- "num_attention_heads": 12,
19
- "num_hidden_layers": 12,
20
  "pad_token_id": 1,
21
- "projection_dim": 768,
22
  "torch_dtype": "float32",
23
  "transformers_version": "4.46.2",
24
  "vocab_size": 49408
 
1
  {
2
+ "_name_or_path": "stabilityai/stable-diffusion-2-inpainting",
3
  "architectures": [
4
  "CLIPTextModel"
5
  ],
 
7
  "bos_token_id": 0,
8
  "dropout": 0.0,
9
  "eos_token_id": 2,
10
+ "hidden_act": "gelu",
11
+ "hidden_size": 1024,
12
  "initializer_factor": 1.0,
13
  "initializer_range": 0.02,
14
+ "intermediate_size": 4096,
15
  "layer_norm_eps": 1e-05,
16
  "max_position_embeddings": 77,
17
  "model_type": "clip_text_model",
18
+ "num_attention_heads": 16,
19
+ "num_hidden_layers": 23,
20
  "pad_token_id": 1,
21
+ "projection_dim": 512,
22
  "torch_dtype": "float32",
23
  "transformers_version": "4.46.2",
24
  "vocab_size": 49408
text_encoder/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:778d02eb9e707c3fbaae0b67b79ea0d1399b52e624fb634f2f19375ae7c047c3
3
- size 492265168
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67e013543d4fac905c882e2993d86a2d454ee69dc9e8f37c0c23d33a48959d15
3
+ size 1361596304
tokenizer/special_tokens_map.json CHANGED
@@ -13,7 +13,7 @@
13
  "rstrip": false,
14
  "single_word": false
15
  },
16
- "pad_token": "<|endoftext|>",
17
  "unk_token": {
18
  "content": "<|endoftext|>",
19
  "lstrip": false,
 
13
  "rstrip": false,
14
  "single_word": false
15
  },
16
+ "pad_token": "!",
17
  "unk_token": {
18
  "content": "<|endoftext|>",
19
  "lstrip": false,
tokenizer/tokenizer_config.json CHANGED
@@ -1,6 +1,14 @@
1
  {
2
  "add_prefix_space": false,
3
  "added_tokens_decoder": {
 
 
 
 
 
 
 
 
4
  "49406": {
5
  "content": "<|startoftext|>",
6
  "lstrip": false,
@@ -24,7 +32,7 @@
24
  "eos_token": "<|endoftext|>",
25
  "errors": "replace",
26
  "model_max_length": 77,
27
- "pad_token": "<|endoftext|>",
28
  "tokenizer_class": "CLIPTokenizer",
29
  "unk_token": "<|endoftext|>"
30
  }
 
1
  {
2
  "add_prefix_space": false,
3
  "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "!",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
  "49406": {
13
  "content": "<|startoftext|>",
14
  "lstrip": false,
 
32
  "eos_token": "<|endoftext|>",
33
  "errors": "replace",
34
  "model_max_length": 77,
35
+ "pad_token": "!",
36
  "tokenizer_class": "CLIPTokenizer",
37
  "unk_token": "<|endoftext|>"
38
  }
unet/config.json CHANGED
@@ -1,12 +1,17 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.31.0",
4
- "_name_or_path": "runwayml/stable-diffusion-inpainting",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
7
  "addition_embed_type_num_heads": 64,
8
  "addition_time_embed_dim": null,
9
- "attention_head_dim": 8,
 
 
 
 
 
10
  "attention_type": "default",
11
  "block_out_channels": [
12
  320,
@@ -19,7 +24,7 @@
19
  "class_embeddings_concat": false,
20
  "conv_in_kernel": 3,
21
  "conv_out_kernel": 3,
22
- "cross_attention_dim": 768,
23
  "cross_attention_norm": null,
24
  "down_block_types": [
25
  "CrossAttnDownBlock2D",
@@ -64,5 +69,5 @@
64
  "CrossAttnUpBlock2D"
65
  ],
66
  "upcast_attention": false,
67
- "use_linear_projection": false
68
  }
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.31.0",
4
+ "_name_or_path": "stabilityai/stable-diffusion-2-inpainting",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
7
  "addition_embed_type_num_heads": 64,
8
  "addition_time_embed_dim": null,
9
+ "attention_head_dim": [
10
+ 5,
11
+ 10,
12
+ 20,
13
+ 20
14
+ ],
15
  "attention_type": "default",
16
  "block_out_channels": [
17
  320,
 
24
  "class_embeddings_concat": false,
25
  "conv_in_kernel": 3,
26
  "conv_out_kernel": 3,
27
+ "cross_attention_dim": 1024,
28
  "cross_attention_norm": null,
29
  "down_block_types": [
30
  "CrossAttnDownBlock2D",
 
69
  "CrossAttnUpBlock2D"
70
  ],
71
  "upcast_attention": false,
72
+ "use_linear_projection": true
73
  }
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f3e2e7da863390677c6e5b8b6b96f338fa3b1f6b1a551d6b8b2d997e847ad35d
3
- size 3438225152
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:804b57b7ec49e32bc890848db36bd0113da5ea3cd43b84f22ca8178a5d0b927f
3
+ size 3463784120
vae/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "AutoencoderKL",
3
  "_diffusers_version": "0.31.0",
4
- "_name_or_path": "C:\\Users\\bsmolka\\.cache\\huggingface\\hub\\models--runwayml--stable-diffusion-inpainting\\snapshots\\8a4288a76071f7280aedbdb3253bdb9e9d5d84bb\\vae",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,
@@ -24,7 +24,7 @@
24
  "mid_block_add_attention": true,
25
  "norm_num_groups": 32,
26
  "out_channels": 3,
27
- "sample_size": 256,
28
  "scaling_factor": 0.18215,
29
  "shift_factor": null,
30
  "up_block_types": [
 
1
  {
2
  "_class_name": "AutoencoderKL",
3
  "_diffusers_version": "0.31.0",
4
+ "_name_or_path": "C:\\Users\\bsmolka\\.cache\\huggingface\\hub\\models--stabilityai--stable-diffusion-2-inpainting\\snapshots\\81a84f49b15956b60b4272a405ad3daef3da4590\\vae",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,
 
24
  "mid_block_add_attention": true,
25
  "norm_num_groups": 32,
26
  "out_channels": 3,
27
+ "sample_size": 512,
28
  "scaling_factor": 0.18215,
29
  "shift_factor": null,
30
  "up_block_types": [
vae/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b4d2b5932bb4151e54e694fd31ccf51fca908223c9485bd56cd0e1d83ad94c49
3
  size 334643268
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2aa1f43011b553a4cba7f37456465cdbd48aab7b54b9348b890e8058ea7683ec
3
  size 334643268