dacorvo HF Staff commited on
Commit
1442e23
·
verified ·
1 Parent(s): 20724f7

Synchronizing local compiler cache.

Browse files
neuronxcc-2.17.194.0+d312836f/0_REGISTRY/0.2.1/stable-diffusion/stabilityai/stable-diffusion-2-1/ee34e363190e7c6310a6.json ADDED
@@ -0,0 +1,127 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_entry_class": "MultiModelCacheEntry",
3
+ "_model_id": "stabilityai/stable-diffusion-2-1",
4
+ "_task": null,
5
+ "text_encoder": {
6
+ "architectures": [
7
+ "CLIPTextModel"
8
+ ],
9
+ "attention_dropout": 0.0,
10
+ "dropout": 0.0,
11
+ "hidden_act": "gelu",
12
+ "hidden_size": 1024,
13
+ "initializer_factor": 1.0,
14
+ "initializer_range": 0.02,
15
+ "intermediate_size": 4096,
16
+ "layer_norm_eps": 1e-05,
17
+ "max_position_embeddings": 77,
18
+ "model_type": "clip_text_model",
19
+ "neuron": {
20
+ "auto_cast": "matmul",
21
+ "auto_cast_type": "bf16",
22
+ "compiler_type": "neuronx-cc",
23
+ "compiler_version": "2.17.194.0+d312836f",
24
+ "dynamic_batch_size": false,
25
+ "inline_weights_to_neff": false,
26
+ "optlevel": "2",
27
+ "output_attentions": false,
28
+ "output_hidden_states": false,
29
+ "static_batch_size": 1,
30
+ "static_sequence_length": 77,
31
+ "task": "feature-extraction",
32
+ "tensor_parallel_size": 1
33
+ },
34
+ "num_attention_heads": 16,
35
+ "num_hidden_layers": 23,
36
+ "vocab_size": 49408
37
+ },
38
+ "unet": {
39
+ "_class_name": "UNet2DConditionModel",
40
+ "act_fn": "silu",
41
+ "addition_embed_type": null,
42
+ "addition_embed_type_num_heads": 64,
43
+ "addition_time_embed_dim": null,
44
+ "attention_head_dim": [
45
+ 5,
46
+ 10,
47
+ 20,
48
+ 20
49
+ ],
50
+ "attention_type": "default",
51
+ "block_out_channels": [
52
+ 320,
53
+ 640,
54
+ 1280,
55
+ 1280
56
+ ],
57
+ "center_input_sample": false,
58
+ "class_embed_type": null,
59
+ "class_embeddings_concat": false,
60
+ "conv_in_kernel": 3,
61
+ "conv_out_kernel": 3,
62
+ "cross_attention_dim": 1024,
63
+ "cross_attention_norm": null,
64
+ "down_block_types": [
65
+ "CrossAttnDownBlock2D",
66
+ "CrossAttnDownBlock2D",
67
+ "CrossAttnDownBlock2D",
68
+ "DownBlock2D"
69
+ ],
70
+ "downsample_padding": 1,
71
+ "dropout": 0.0,
72
+ "dual_cross_attention": false,
73
+ "encoder_hid_dim": null,
74
+ "encoder_hid_dim_type": null,
75
+ "flip_sin_to_cos": true,
76
+ "freq_shift": 0,
77
+ "in_channels": 4,
78
+ "layers_per_block": 2,
79
+ "mid_block_only_cross_attention": null,
80
+ "mid_block_scale_factor": 1,
81
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
82
+ "neuron": {
83
+ "auto_cast": "matmul",
84
+ "auto_cast_type": "bf16",
85
+ "compiler_type": "neuronx-cc",
86
+ "compiler_version": "2.17.194.0+d312836f",
87
+ "dynamic_batch_size": false,
88
+ "inline_weights_to_neff": false,
89
+ "optlevel": "2",
90
+ "output_attentions": false,
91
+ "output_hidden_states": false,
92
+ "static_batch_size": 1,
93
+ "static_height": 64,
94
+ "static_num_channels": 4,
95
+ "static_sequence_length": 77,
96
+ "static_vae_scale_factor": 8,
97
+ "static_width": 64,
98
+ "task": "semantic-segmentation",
99
+ "tensor_parallel_size": 1
100
+ },
101
+ "norm_eps": 1e-05,
102
+ "norm_num_groups": 32,
103
+ "num_attention_heads": null,
104
+ "num_class_embeds": null,
105
+ "only_cross_attention": false,
106
+ "out_channels": 4,
107
+ "projection_class_embeddings_input_dim": null,
108
+ "resnet_out_scale_factor": 1.0,
109
+ "resnet_skip_time_act": false,
110
+ "resnet_time_scale_shift": "default",
111
+ "reverse_transformer_layers_per_block": null,
112
+ "time_cond_proj_dim": null,
113
+ "time_embedding_act_fn": null,
114
+ "time_embedding_dim": null,
115
+ "time_embedding_type": "positional",
116
+ "timestep_post_act": null,
117
+ "transformer_layers_per_block": 1,
118
+ "up_block_types": [
119
+ "UpBlock2D",
120
+ "CrossAttnUpBlock2D",
121
+ "CrossAttnUpBlock2D",
122
+ "CrossAttnUpBlock2D"
123
+ ],
124
+ "upcast_attention": true,
125
+ "use_linear_projection": true
126
+ }
127
+ }
neuronxcc-2.17.194.0+d312836f/MODULE_ee34e363190e7c6310a6/text_encoder/model.neuron CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:83b53cb0ab9cc91c116384887636af6c94609ec89880bcbdc97903640caa7d63
3
  size 1372305465
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8aaec2056307c7140cc100a0bd6ef916ee5d103a6336af91d8d1add4ded807e3
3
  size 1372305465
neuronxcc-2.17.194.0+d312836f/MODULE_ee34e363190e7c6310a6/unet/config.json CHANGED
@@ -3,32 +3,32 @@
3
  "_commit_hash": null,
4
  "_diffusers_version": "0.30.3",
5
  "_use_default_values": [
6
- "class_embeddings_concat",
7
- "addition_embed_type",
8
- "mid_block_only_cross_attention",
9
  "mid_block_type",
 
10
  "class_embed_type",
 
 
 
11
  "projection_class_embeddings_input_dim",
12
- "time_embedding_dim",
13
- "encoder_hid_dim",
14
- "timestep_post_act",
15
- "conv_in_kernel",
16
- "resnet_out_scale_factor",
17
- "dropout",
18
  "time_embedding_act_fn",
 
 
19
  "resnet_time_scale_shift",
20
- "resnet_skip_time_act",
21
- "num_attention_heads",
22
- "encoder_hid_dim_type",
23
- "transformer_layers_per_block",
24
- "conv_out_kernel",
25
  "attention_type",
26
- "time_embedding_type",
27
- "addition_time_embed_dim",
 
28
  "reverse_transformer_layers_per_block",
 
29
  "time_cond_proj_dim",
30
- "cross_attention_norm",
31
- "addition_embed_type_num_heads"
 
 
32
  ],
33
  "act_fn": "silu",
34
  "addition_embed_type": null,
 
3
  "_commit_hash": null,
4
  "_diffusers_version": "0.30.3",
5
  "_use_default_values": [
 
 
 
6
  "mid_block_type",
7
+ "timestep_post_act",
8
  "class_embed_type",
9
+ "addition_embed_type_num_heads",
10
+ "time_embedding_type",
11
+ "conv_out_kernel",
12
  "projection_class_embeddings_input_dim",
13
+ "transformer_layers_per_block",
14
+ "mid_block_only_cross_attention",
15
+ "encoder_hid_dim_type",
16
+ "addition_embed_type",
 
 
17
  "time_embedding_act_fn",
18
+ "resnet_out_scale_factor",
19
+ "addition_time_embed_dim",
20
  "resnet_time_scale_shift",
 
 
 
 
 
21
  "attention_type",
22
+ "num_attention_heads",
23
+ "cross_attention_norm",
24
+ "conv_in_kernel",
25
  "reverse_transformer_layers_per_block",
26
+ "resnet_skip_time_act",
27
  "time_cond_proj_dim",
28
+ "encoder_hid_dim",
29
+ "dropout",
30
+ "time_embedding_dim",
31
+ "class_embeddings_concat"
32
  ],
33
  "act_fn": "silu",
34
  "addition_embed_type": null,
neuronxcc-2.17.194.0+d312836f/MODULE_ee34e363190e7c6310a6/unet/model.neuron CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e3d84c257a117da92ee904d812faead592eb13d0be3fa66b6419a9933ff22475
3
  size 3572109069
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a29948a5d490a0aec1763c947bff9b5725ee72a6a61793d1ea296270bc0324de
3
  size 3572109069
neuronxcc-2.17.194.0+d312836f/MODULE_ee34e363190e7c6310a6/vae_decoder/config.json CHANGED
@@ -3,13 +3,13 @@
3
  "_commit_hash": null,
4
  "_diffusers_version": "0.30.3",
5
  "_use_default_values": [
6
- "latents_mean",
7
  "scaling_factor",
8
  "shift_factor",
9
- "use_quant_conv",
10
  "mid_block_add_attention",
11
- "force_upcast",
12
  "latents_std",
 
 
13
  "use_post_quant_conv"
14
  ],
15
  "act_fn": "silu",
 
3
  "_commit_hash": null,
4
  "_diffusers_version": "0.30.3",
5
  "_use_default_values": [
6
+ "use_quant_conv",
7
  "scaling_factor",
8
  "shift_factor",
 
9
  "mid_block_add_attention",
 
10
  "latents_std",
11
+ "latents_mean",
12
+ "force_upcast",
13
  "use_post_quant_conv"
14
  ],
15
  "act_fn": "silu",
neuronxcc-2.17.194.0+d312836f/MODULE_ee34e363190e7c6310a6/vae_decoder/model.neuron CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:de01f9329d79dba386c90acf88b27b1d03899642df99275c6efc1683b9f3f07e
3
  size 367178901
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d06d75627041c8d9ebb0e6e6c923a11c50a9e0567381c78a7de557a7501f9e0
3
  size 367178901
neuronxcc-2.17.194.0+d312836f/MODULE_ee34e363190e7c6310a6/vae_encoder/config.json CHANGED
@@ -3,13 +3,13 @@
3
  "_commit_hash": null,
4
  "_diffusers_version": "0.30.3",
5
  "_use_default_values": [
6
- "latents_mean",
7
  "scaling_factor",
8
  "shift_factor",
9
- "use_quant_conv",
10
  "mid_block_add_attention",
11
- "force_upcast",
12
  "latents_std",
 
 
13
  "use_post_quant_conv"
14
  ],
15
  "act_fn": "silu",
 
3
  "_commit_hash": null,
4
  "_diffusers_version": "0.30.3",
5
  "_use_default_values": [
6
+ "use_quant_conv",
7
  "scaling_factor",
8
  "shift_factor",
 
9
  "mid_block_add_attention",
 
10
  "latents_std",
11
+ "latents_mean",
12
+ "force_upcast",
13
  "use_post_quant_conv"
14
  ],
15
  "act_fn": "silu",
neuronxcc-2.17.194.0+d312836f/MODULE_ee34e363190e7c6310a6/vae_encoder/model.neuron CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d7b1d86dbc1f51a502fc4ffd02f9de38ca0250b8a173d532065c99645b25e611
3
  size 225662483
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:95a74e139d9f8ef63e5f592df2ee84cee6413158d12e9184ca7b73c58652473a
3
  size 225662483