jburtoft commited on
Commit
08e5765
1 Parent(s): 30d4b7e

Upload folder using huggingface_hub (#1)

Browse files

- 86d4261d61a2204a879e99684f27b2ec2c942f4a73d05019a2caaade089a542c (2c261c9d982871c81a75fe704331fbab5225f5ca)
- 199a47e857735e7207922d64170ee3b710ab27367e5629986d2247af12959d68 (ef52310c104ee0a2f0db9eb596a4551ae861b218)
- a96072f8904876dd62f6b4c9d3d071c5d2efe255bdd5777a8e84d218efac6f2b (acd93f13222ddf3af997b5e37a0ef9c50cd067df)
- 6508ba4621e6b3a6a200e399c2593d8cc725e0f32ba4a0e60ebed6700c7f82c8 (0a183905cb7947fc067f2c9ec2064c9b54eac06c)

.gitattributes CHANGED
@@ -33,3 +33,8 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ text_encoder/model.neuron filter=lfs diff=lfs merge=lfs -text
37
+ unet/model.neuron filter=lfs diff=lfs merge=lfs -text
38
+ text_encoder_2/model.neuron filter=lfs diff=lfs merge=lfs -text
39
+ vae_decoder/model.neuron filter=lfs diff=lfs merge=lfs -text
40
+ vae_encoder/model.neuron filter=lfs diff=lfs merge=lfs -text
model_index.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "NeuronStableDiffusionXLPipeline",
3
+ "_diffusers_version": "0.25.0",
4
+ "_name_or_path": "lykon-models/dreamshaper-xl-1-0",
5
+ "force_zeros_for_empty_prompt": true,
6
+ "scheduler": [
7
+ "diffusers",
8
+ "DEISMultistepScheduler"
9
+ ],
10
+ "text_encoder": [
11
+ "optimum",
12
+ "NeuronModelTextEncoder"
13
+ ],
14
+ "text_encoder_2": [
15
+ "optimum",
16
+ "NeuronModelTextEncoder"
17
+ ],
18
+ "tokenizer": [
19
+ "transformers",
20
+ "CLIPTokenizer"
21
+ ],
22
+ "tokenizer_2": [
23
+ "transformers",
24
+ "CLIPTokenizer"
25
+ ],
26
+ "unet": [
27
+ "optimum",
28
+ "NeuronModelUnet"
29
+ ],
30
+ "vae_decoder": [
31
+ "optimum",
32
+ "NeuronModelVaeDecoder"
33
+ ],
34
+ "vae_encoder": [
35
+ "optimum",
36
+ "NeuronModelVaeEncoder"
37
+ ]
38
+ }
scheduler/scheduler_config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DEISMultistepScheduler",
3
+ "_diffusers_version": "0.25.0",
4
+ "algorithm_type": "deis",
5
+ "beta_end": 0.012,
6
+ "beta_schedule": "scaled_linear",
7
+ "beta_start": 0.00085,
8
+ "clip_sample": false,
9
+ "dynamic_thresholding_ratio": 0.995,
10
+ "interpolation_type": "linear",
11
+ "lower_order_final": true,
12
+ "num_train_timesteps": 1000,
13
+ "prediction_type": "epsilon",
14
+ "sample_max_value": 1.0,
15
+ "set_alpha_to_one": false,
16
+ "skip_prk_steps": true,
17
+ "solver_order": 2,
18
+ "solver_type": "logrho",
19
+ "steps_offset": 1,
20
+ "thresholding": false,
21
+ "timestep_spacing": "leading",
22
+ "trained_betas": null,
23
+ "use_karras_sigmas": false
24
+ }
text_encoder/config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/home/ubuntu/.cache/huggingface/hub/models--Lykon--dreamshaper-xl-1-0/snapshots/41e6644752a8c9aa63930e6043c4fd83c7708420/text_encoder",
3
+ "architectures": [
4
+ "CLIPTextModel"
5
+ ],
6
+ "attention_dropout": 0.0,
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
+ "neuron": {
19
+ "auto_cast": "all",
20
+ "auto_cast_type": "bf16",
21
+ "compiler_type": "neuronx-cc",
22
+ "compiler_version": "2.12.68.0+4480452af",
23
+ "disable_fallback": false,
24
+ "disable_fast_relayout": false,
25
+ "dynamic_batch_size": false,
26
+ "input_names": [
27
+ "input_ids"
28
+ ],
29
+ "model_type": "clip-text-model",
30
+ "optlevel": "2",
31
+ "output_attentions": false,
32
+ "output_hidden_states": false,
33
+ "output_names": [
34
+ "last_hidden_state",
35
+ "pooler_output",
36
+ "hidden_states"
37
+ ],
38
+ "static_batch_size": 1,
39
+ "static_num_beams": 1,
40
+ "static_sequence_length": 77
41
+ },
42
+ "num_attention_heads": 12,
43
+ "num_hidden_layers": 12,
44
+ "output_hidden_states": true,
45
+ "pad_token_id": 1,
46
+ "projection_dim": 768,
47
+ "task": "feature-extraction",
48
+ "torch_dtype": "float16",
49
+ "torchscript": true,
50
+ "transformers_version": "4.36.2",
51
+ "vocab_size": 49408
52
+ }
text_encoder/model.neuron ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:98750e8358b1864ea984f44170f1c06979234b67a53d4461b342a8626593b16d
3
+ size 375884546
text_encoder_2/config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/home/ubuntu/.cache/huggingface/hub/models--Lykon--dreamshaper-xl-1-0/snapshots/41e6644752a8c9aa63930e6043c4fd83c7708420/text_encoder_2",
3
+ "architectures": [
4
+ "CLIPTextModelWithProjection"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 0,
8
+ "dropout": 0.0,
9
+ "eos_token_id": 2,
10
+ "hidden_act": "gelu",
11
+ "hidden_size": 1280,
12
+ "initializer_factor": 1.0,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 5120,
15
+ "layer_norm_eps": 1e-05,
16
+ "max_position_embeddings": 77,
17
+ "model_type": "clip_text_model",
18
+ "neuron": {
19
+ "auto_cast": "all",
20
+ "auto_cast_type": "bf16",
21
+ "compiler_type": "neuronx-cc",
22
+ "compiler_version": "2.12.68.0+4480452af",
23
+ "disable_fallback": false,
24
+ "disable_fast_relayout": false,
25
+ "dynamic_batch_size": false,
26
+ "input_names": [
27
+ "input_ids"
28
+ ],
29
+ "model_type": "clip-text-model",
30
+ "optlevel": "2",
31
+ "output_attentions": false,
32
+ "output_hidden_states": false,
33
+ "output_names": [
34
+ "text_embeds",
35
+ "last_hidden_state",
36
+ "hidden_states"
37
+ ],
38
+ "static_batch_size": 1,
39
+ "static_num_beams": 1,
40
+ "static_sequence_length": 77
41
+ },
42
+ "num_attention_heads": 20,
43
+ "num_hidden_layers": 32,
44
+ "output_hidden_states": true,
45
+ "pad_token_id": 1,
46
+ "projection_dim": 1280,
47
+ "task": "feature-extraction",
48
+ "torch_dtype": "float16",
49
+ "torchscript": true,
50
+ "transformers_version": "4.36.2",
51
+ "vocab_size": 49408
52
+ }
text_encoder_2/model.neuron ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32a280df8cafb7abdcfba4a08b1534220ebfbb6b493ab13da8f93ba7328da101
3
+ size 1788954872
tokenizer/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer/special_tokens_map.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<|startoftext|>",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "<|endoftext|>",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<|endoftext|>",
18
+ "lstrip": false,
19
+ "normalized": true,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "unk_token": {
24
+ "content": "<|endoftext|>",
25
+ "lstrip": false,
26
+ "normalized": true,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ }
30
+ }
tokenizer/tokenizer_config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "49406": {
5
+ "content": "<|startoftext|>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "49407": {
13
+ "content": "<|endoftext|>",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ }
20
+ },
21
+ "bos_token": "<|startoftext|>",
22
+ "clean_up_tokenization_spaces": true,
23
+ "do_lower_case": true,
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
+ }
tokenizer/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_2/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_2/special_tokens_map.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<|startoftext|>",
4
+ "lstrip": false,
5
+ "normalized": true,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "<|endoftext|>",
11
+ "lstrip": false,
12
+ "normalized": true,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "!",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "unk_token": {
24
+ "content": "<|endoftext|>",
25
+ "lstrip": false,
26
+ "normalized": true,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ }
30
+ }
tokenizer_2/tokenizer_config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "49407": {
21
+ "content": "<|endoftext|>",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ }
28
+ },
29
+ "bos_token": "<|startoftext|>",
30
+ "clean_up_tokenization_spaces": true,
31
+ "do_lower_case": true,
32
+ "eos_token": "<|endoftext|>",
33
+ "errors": "replace",
34
+ "model_max_length": 77,
35
+ "pad_token": "!",
36
+ "tokenizer_class": "CLIPTokenizer",
37
+ "unk_token": "<|endoftext|>"
38
+ }
tokenizer_2/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
unet/config.json ADDED
@@ -0,0 +1,109 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_commit_hash": null,
4
+ "_diffusers_version": "0.25.0",
5
+ "_name_or_path": "/home/ubuntu/.cache/huggingface/hub/models--Lykon--dreamshaper-xl-1-0/snapshots/41e6644752a8c9aa63930e6043c4fd83c7708420/unet",
6
+ "_use_default_values": [
7
+ "dropout",
8
+ "reverse_transformer_layers_per_block"
9
+ ],
10
+ "act_fn": "silu",
11
+ "addition_embed_type": "text_time",
12
+ "addition_embed_type_num_heads": 64,
13
+ "addition_time_embed_dim": 256,
14
+ "attention_head_dim": [
15
+ 5,
16
+ 10,
17
+ 20
18
+ ],
19
+ "attention_type": "default",
20
+ "block_out_channels": [
21
+ 320,
22
+ 640,
23
+ 1280
24
+ ],
25
+ "center_input_sample": false,
26
+ "class_embed_type": null,
27
+ "class_embeddings_concat": false,
28
+ "conv_in_kernel": 3,
29
+ "conv_out_kernel": 3,
30
+ "cross_attention_dim": 2048,
31
+ "cross_attention_norm": null,
32
+ "down_block_types": [
33
+ "DownBlock2D",
34
+ "CrossAttnDownBlock2D",
35
+ "CrossAttnDownBlock2D"
36
+ ],
37
+ "downsample_padding": 1,
38
+ "dropout": 0.0,
39
+ "dual_cross_attention": false,
40
+ "encoder_hid_dim": null,
41
+ "encoder_hid_dim_type": null,
42
+ "flip_sin_to_cos": true,
43
+ "freq_shift": 0,
44
+ "in_channels": 4,
45
+ "layers_per_block": 2,
46
+ "mid_block_only_cross_attention": null,
47
+ "mid_block_scale_factor": 1,
48
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
49
+ "neuron": {
50
+ "auto_cast": "all",
51
+ "auto_cast_type": "bf16",
52
+ "compiler_type": "neuronx-cc",
53
+ "compiler_version": "2.12.68.0+4480452af",
54
+ "disable_fallback": false,
55
+ "disable_fast_relayout": false,
56
+ "dynamic_batch_size": false,
57
+ "input_names": [
58
+ "sample",
59
+ "timestep",
60
+ "encoder_hidden_states",
61
+ "text_embeds",
62
+ "time_ids"
63
+ ],
64
+ "model_type": "unet",
65
+ "optlevel": "2",
66
+ "output_attentions": false,
67
+ "output_hidden_states": false,
68
+ "output_names": [
69
+ "sample"
70
+ ],
71
+ "static_batch_size": 1,
72
+ "static_height": 128,
73
+ "static_num_beams": 1,
74
+ "static_num_channels": 4,
75
+ "static_sequence_length": 77,
76
+ "static_width": 128
77
+ },
78
+ "norm_eps": 1e-05,
79
+ "norm_num_groups": 32,
80
+ "num_attention_heads": null,
81
+ "num_class_embeds": null,
82
+ "only_cross_attention": false,
83
+ "out_channels": 4,
84
+ "projection_class_embeddings_input_dim": 2816,
85
+ "resnet_out_scale_factor": 1.0,
86
+ "resnet_skip_time_act": false,
87
+ "resnet_time_scale_shift": "default",
88
+ "reverse_transformer_layers_per_block": null,
89
+ "sample_size": 128,
90
+ "task": "semantic-segmentation",
91
+ "time_cond_proj_dim": null,
92
+ "time_embedding_act_fn": null,
93
+ "time_embedding_dim": null,
94
+ "time_embedding_type": "positional",
95
+ "timestep_post_act": null,
96
+ "transformer_layers_per_block": [
97
+ 1,
98
+ 2,
99
+ 10
100
+ ],
101
+ "transformers_version": null,
102
+ "up_block_types": [
103
+ "CrossAttnUpBlock2D",
104
+ "CrossAttnUpBlock2D",
105
+ "UpBlock2D"
106
+ ],
107
+ "upcast_attention": null,
108
+ "use_linear_projection": true
109
+ }
unet/model.neuron ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e2ae64034e8d13933e24c98fda33a9e092a605ed826869bd7f9f28a56f29f9c
3
+ size 4175089276
vae_decoder/config.json ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "AutoencoderKL",
3
+ "_commit_hash": null,
4
+ "_diffusers_version": "0.25.0",
5
+ "_name_or_path": "/home/ubuntu/.cache/huggingface/hub/models--Lykon--dreamshaper-xl-1-0/snapshots/41e6644752a8c9aa63930e6043c4fd83c7708420/vae",
6
+ "act_fn": "silu",
7
+ "block_out_channels": [
8
+ 128,
9
+ 256,
10
+ 512,
11
+ 512
12
+ ],
13
+ "down_block_types": [
14
+ "DownEncoderBlock2D",
15
+ "DownEncoderBlock2D",
16
+ "DownEncoderBlock2D",
17
+ "DownEncoderBlock2D"
18
+ ],
19
+ "force_upcast": true,
20
+ "in_channels": 3,
21
+ "latent_channels": 4,
22
+ "layers_per_block": 2,
23
+ "neuron": {
24
+ "auto_cast": "all",
25
+ "auto_cast_type": "bf16",
26
+ "compiler_type": "neuronx-cc",
27
+ "compiler_version": "2.12.68.0+4480452af",
28
+ "disable_fallback": false,
29
+ "disable_fast_relayout": false,
30
+ "dynamic_batch_size": false,
31
+ "input_names": [
32
+ "latent_sample"
33
+ ],
34
+ "model_type": "vae-decoder",
35
+ "optlevel": "2",
36
+ "output_attentions": false,
37
+ "output_hidden_states": false,
38
+ "output_names": [
39
+ "sample"
40
+ ],
41
+ "static_batch_size": 1,
42
+ "static_height": 128,
43
+ "static_num_beams": 1,
44
+ "static_num_channels": 4,
45
+ "static_width": 128
46
+ },
47
+ "norm_num_groups": 32,
48
+ "out_channels": 3,
49
+ "sample_size": 1024,
50
+ "scaling_factor": 0.13025,
51
+ "task": "semantic-segmentation",
52
+ "transformers_version": null,
53
+ "up_block_types": [
54
+ "UpDecoderBlock2D",
55
+ "UpDecoderBlock2D",
56
+ "UpDecoderBlock2D",
57
+ "UpDecoderBlock2D"
58
+ ]
59
+ }
vae_decoder/model.neuron ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e8040811aa76bebf4e8667b4341a66dab37a94f1c6d50ffa82002dae09b23beb
3
+ size 811576830
vae_encoder/config.json ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "AutoencoderKL",
3
+ "_commit_hash": null,
4
+ "_diffusers_version": "0.25.0",
5
+ "_name_or_path": "/home/ubuntu/.cache/huggingface/hub/models--Lykon--dreamshaper-xl-1-0/snapshots/41e6644752a8c9aa63930e6043c4fd83c7708420/vae",
6
+ "act_fn": "silu",
7
+ "block_out_channels": [
8
+ 128,
9
+ 256,
10
+ 512,
11
+ 512
12
+ ],
13
+ "down_block_types": [
14
+ "DownEncoderBlock2D",
15
+ "DownEncoderBlock2D",
16
+ "DownEncoderBlock2D",
17
+ "DownEncoderBlock2D"
18
+ ],
19
+ "force_upcast": true,
20
+ "in_channels": 3,
21
+ "latent_channels": 4,
22
+ "layers_per_block": 2,
23
+ "neuron": {
24
+ "auto_cast": "all",
25
+ "auto_cast_type": "bf16",
26
+ "compiler_type": "neuronx-cc",
27
+ "compiler_version": "2.12.68.0+4480452af",
28
+ "disable_fallback": false,
29
+ "disable_fast_relayout": false,
30
+ "dynamic_batch_size": false,
31
+ "input_names": [
32
+ "sample"
33
+ ],
34
+ "model_type": "vae-encoder",
35
+ "optlevel": "2",
36
+ "output_attentions": false,
37
+ "output_hidden_states": false,
38
+ "output_names": [
39
+ "latent_sample"
40
+ ],
41
+ "static_batch_size": 1,
42
+ "static_height": 1024,
43
+ "static_num_beams": 1,
44
+ "static_num_channels": 3,
45
+ "static_width": 1024
46
+ },
47
+ "norm_num_groups": 32,
48
+ "out_channels": 3,
49
+ "sample_size": 1024,
50
+ "scaling_factor": 0.13025,
51
+ "task": "semantic-segmentation",
52
+ "transformers_version": null,
53
+ "up_block_types": [
54
+ "UpDecoderBlock2D",
55
+ "UpDecoderBlock2D",
56
+ "UpDecoderBlock2D",
57
+ "UpDecoderBlock2D"
58
+ ]
59
+ }
vae_encoder/model.neuron ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d809a457b0d9c63c755290ea4920867c41dd0bc5622a2dce06fee0fbd94003af
3
+ size 398106110