pradyumnachari commited on
Commit
2e980cd
1 Parent(s): 5ede9e4

Adding `diffusers` weights of this model

Browse files

This PR adds fp32 and fp16 weights in PyTorch and safetensors format to stabilityai/stable-diffusion-2-1-base

model_index.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
  "_class_name": "StableDiffusionPipeline",
3
- "_diffusers_version": "0.10.0.dev0",
4
  "feature_extractor": [
5
- "transformers",
6
- "CLIPImageProcessor"
7
  ],
8
  "requires_safety_checker": false,
9
  "safety_checker": [
@@ -12,7 +12,7 @@
12
  ],
13
  "scheduler": [
14
  "diffusers",
15
- "PNDMScheduler"
16
  ],
17
  "text_encoder": [
18
  "transformers",
 
1
  {
2
  "_class_name": "StableDiffusionPipeline",
3
+ "_diffusers_version": "0.18.0.dev0",
4
  "feature_extractor": [
5
+ null,
6
+ null
7
  ],
8
  "requires_safety_checker": false,
9
  "safety_checker": [
 
12
  ],
13
  "scheduler": [
14
  "diffusers",
15
+ "DPMSolverMultistepScheduler"
16
  ],
17
  "text_encoder": [
18
  "transformers",
scheduler/scheduler_config.json CHANGED
@@ -1,14 +1,26 @@
1
  {
2
- "_class_name": "PNDMScheduler",
3
- "_diffusers_version": "0.10.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
  }
 
1
  {
2
+ "_class_name": "DPMSolverMultistepScheduler",
3
+ "_diffusers_version": "0.18.0.dev0",
4
+ "algorithm_type": "dpmsolver++",
5
  "beta_end": 0.012,
6
  "beta_schedule": "scaled_linear",
7
  "beta_start": 0.00085,
8
  "clip_sample": false,
9
+ "clip_sample_range": 1.0,
10
+ "dynamic_thresholding_ratio": 0.995,
11
+ "lambda_min_clipped": -Infinity,
12
+ "lower_order_final": true,
13
  "num_train_timesteps": 1000,
14
  "prediction_type": "epsilon",
15
+ "rescale_betas_zero_snr": false,
16
+ "sample_max_value": 1.0,
17
  "set_alpha_to_one": false,
18
+ "solver_order": 2,
19
+ "solver_type": "midpoint",
20
  "steps_offset": 1,
21
+ "thresholding": false,
22
+ "timestep_spacing": "leading",
23
+ "trained_betas": null,
24
+ "use_karras_sigmas": false,
25
+ "variance_type": null
26
  }
text_encoder/config.json CHANGED
@@ -19,7 +19,7 @@
19
  "num_hidden_layers": 23,
20
  "pad_token_id": 1,
21
  "projection_dim": 512,
22
- "torch_dtype": "float32",
23
- "transformers_version": "4.25.0.dev0",
24
  "vocab_size": 49408
25
  }
 
19
  "num_hidden_layers": 23,
20
  "pad_token_id": 1,
21
  "projection_dim": 512,
22
+ "torch_dtype": "float16",
23
+ "transformers_version": "4.30.2",
24
  "vocab_size": 49408
25
  }
text_encoder/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cce6febb0b6d876ee5eb24af35e27e764eb4f9b1d0b7c026c8c3333d4cfc916c
3
- size 1361597018
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1e4aa519f64dc6386f88221a66c106a09fa027b47a20cc0e126687695f2a6669
3
+ size 1361597016
text_encoder/pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c3e254d7b61353497ea0be2c4013df4ea8f739ee88cffa0ba58cd085459ed565
3
- size 1361671895
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2188379b05015f531d61503e714234d00a64939792f3098b324e516547f0194f
3
+ size 1361674657
tokenizer/tokenizer_config.json CHANGED
@@ -8,6 +8,7 @@
8
  "rstrip": false,
9
  "single_word": false
10
  },
 
11
  "do_lower_case": true,
12
  "eos_token": {
13
  "__type": "AddedToken",
@@ -19,9 +20,7 @@
19
  },
20
  "errors": "replace",
21
  "model_max_length": 77,
22
- "name_or_path": "stabilityai/stable-diffusion-2",
23
  "pad_token": "<|endoftext|>",
24
- "special_tokens_map_file": "./special_tokens_map.json",
25
  "tokenizer_class": "CLIPTokenizer",
26
  "unk_token": {
27
  "__type": "AddedToken",
 
8
  "rstrip": false,
9
  "single_word": false
10
  },
11
+ "clean_up_tokenization_spaces": true,
12
  "do_lower_case": true,
13
  "eos_token": {
14
  "__type": "AddedToken",
 
20
  },
21
  "errors": "replace",
22
  "model_max_length": 77,
 
23
  "pad_token": "<|endoftext|>",
 
24
  "tokenizer_class": "CLIPTokenizer",
25
  "unk_token": {
26
  "__type": "AddedToken",
unet/config.json CHANGED
@@ -1,7 +1,9 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
- "_diffusers_version": "0.10.0.dev0",
4
  "act_fn": "silu",
 
 
5
  "attention_head_dim": [
6
  5,
7
  10,
@@ -15,7 +17,12 @@
15
  1280
16
  ],
17
  "center_input_sample": false,
 
 
 
 
18
  "cross_attention_dim": 1024,
 
19
  "down_block_types": [
20
  "CrossAttnDownBlock2D",
21
  "CrossAttnDownBlock2D",
@@ -24,22 +31,37 @@
24
  ],
25
  "downsample_padding": 1,
26
  "dual_cross_attention": false,
 
 
27
  "flip_sin_to_cos": true,
28
  "freq_shift": 0,
29
  "in_channels": 4,
30
  "layers_per_block": 2,
 
31
  "mid_block_scale_factor": 1,
 
32
  "norm_eps": 1e-05,
33
  "norm_num_groups": 32,
 
34
  "num_class_embeds": null,
35
  "only_cross_attention": false,
36
  "out_channels": 4,
 
 
 
 
37
  "sample_size": 64,
 
 
 
 
 
38
  "up_block_types": [
39
  "UpBlock2D",
40
  "CrossAttnUpBlock2D",
41
  "CrossAttnUpBlock2D",
42
  "CrossAttnUpBlock2D"
43
  ],
 
44
  "use_linear_projection": true
45
  }
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.18.0.dev0",
4
  "act_fn": "silu",
5
+ "addition_embed_type": null,
6
+ "addition_embed_type_num_heads": 64,
7
  "attention_head_dim": [
8
  5,
9
  10,
 
17
  1280
18
  ],
19
  "center_input_sample": false,
20
+ "class_embed_type": null,
21
+ "class_embeddings_concat": false,
22
+ "conv_in_kernel": 3,
23
+ "conv_out_kernel": 3,
24
  "cross_attention_dim": 1024,
25
+ "cross_attention_norm": null,
26
  "down_block_types": [
27
  "CrossAttnDownBlock2D",
28
  "CrossAttnDownBlock2D",
 
31
  ],
32
  "downsample_padding": 1,
33
  "dual_cross_attention": false,
34
+ "encoder_hid_dim": null,
35
+ "encoder_hid_dim_type": null,
36
  "flip_sin_to_cos": true,
37
  "freq_shift": 0,
38
  "in_channels": 4,
39
  "layers_per_block": 2,
40
+ "mid_block_only_cross_attention": null,
41
  "mid_block_scale_factor": 1,
42
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
43
  "norm_eps": 1e-05,
44
  "norm_num_groups": 32,
45
+ "num_attention_heads": null,
46
  "num_class_embeds": null,
47
  "only_cross_attention": false,
48
  "out_channels": 4,
49
+ "projection_class_embeddings_input_dim": null,
50
+ "resnet_out_scale_factor": 1.0,
51
+ "resnet_skip_time_act": false,
52
+ "resnet_time_scale_shift": "default",
53
  "sample_size": 64,
54
+ "time_cond_proj_dim": null,
55
+ "time_embedding_act_fn": null,
56
+ "time_embedding_dim": null,
57
+ "time_embedding_type": "positional",
58
+ "timestep_post_act": null,
59
  "up_block_types": [
60
  "UpBlock2D",
61
  "CrossAttnUpBlock2D",
62
  "CrossAttnUpBlock2D",
63
  "CrossAttnUpBlock2D"
64
  ],
65
+ "upcast_attention": null,
66
  "use_linear_projection": true
67
  }
unet/diffusion_pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a434f8fdbb51f080ac57a4377281241467c4efc1417b27646ffae989644de60c
3
- size 3463913445
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e45229700c1cb952444ecfe5a22126cdfb1ee72a5352ab5d38812b36277d84b
3
+ size 3463925093
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6dfae3e5f7d459b50f4b0850ead945972c75bb0e1897628933e169eb43974214
3
- size 3463726498
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f349f136057c7ddefbf454ba56b1f9a84ae15436b9b2e677a6590aa966f1db64
3
+ size 3463726504
vae/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "AutoencoderKL",
3
- "_diffusers_version": "0.10.0.dev0",
4
  "act_fn": "silu",
5
  "block_out_channels": [
6
  128,
@@ -19,7 +19,8 @@
19
  "layers_per_block": 2,
20
  "norm_num_groups": 32,
21
  "out_channels": 3,
22
- "sample_size": 768,
 
23
  "up_block_types": [
24
  "UpDecoderBlock2D",
25
  "UpDecoderBlock2D",
 
1
  {
2
  "_class_name": "AutoencoderKL",
3
+ "_diffusers_version": "0.18.0.dev0",
4
  "act_fn": "silu",
5
  "block_out_channels": [
6
  128,
 
19
  "layers_per_block": 2,
20
  "norm_num_groups": 32,
21
  "out_channels": 3,
22
+ "sample_size": 512,
23
+ "scaling_factor": 0.18215,
24
  "up_block_types": [
25
  "UpDecoderBlock2D",
26
  "UpDecoderBlock2D",
vae/diffusion_pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1b4889b6b1d4ce7ae320a02dedaeff1780ad77d415ea0d744b476155c6377ddc
3
- size 334707217
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b01618945554d9840701d3453d4a9fe3db0db090164a5ed6305641306285b6f
3
+ size 334712113
vae/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a1d993488569e928462932c8c38a0760b874d166399b14414135bd9c42df5815
3
- size 334643276
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2aa1f43011b553a4cba7f37456465cdbd48aab7b54b9348b890e8058ea7683ec
3
+ size 334643268