Hugging Face's logo

optimum
/
tiny-stable-diffusion-neuronx

Model card Files Community
Jingya HF Staff commited on
Commit
fc69b23
·
1 Parent(s): 0f5606d

update

Browse files
Files changed (8) hide show
  1. README.md +1 -1
  2. model_index.json +8 -12
  3. text_encoder/model.neuron +2 -2
  4. unet/model.neuron +2 -2
  5. vae_decoder/config.json +2 -2
  6. vae_decoder/model.neuron +2 -2
  7. {vae_post_quant_conv → vae_encoder}/config.json +3 -3
  8. {vae_post_quant_conv → vae_encoder}/model.neuron +2 -2
README.md CHANGED
@@ -13,4 +13,4 @@ stable_diffusion.save_pretrained(save_directory)
13
 
14
  prompt = "sailing ship in storm by Leonardo da Vinci"
15
  image = stable_diffusion(prompt).images[0]
16
- ```
 
13
 
14
  prompt = "sailing ship in storm by Leonardo da Vinci"
15
  image = stable_diffusion(prompt).images[0]
16
+ ```
model_index.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_class_name": "NeuronStableDiffusionPipeline",
3
  "_diffusers_version": "0.17.1",
4
  "feature_extractor": [
5
  "transformers",
@@ -15,23 +15,19 @@
15
  "PNDMScheduler"
16
  ],
17
  "text_encoder": [
18
- "optimum",
19
- "NeuronModelTextEncoder"
20
  ],
21
  "tokenizer": [
22
  "transformers",
23
  "CLIPTokenizer"
24
  ],
25
  "unet": [
26
- "optimum",
27
- "NeuronModelUnet"
28
- ],
29
- "vae_decoder": [
30
- "optimum",
31
- "NeuronModelVaeDecoder"
32
  ],
33
- "vae_post_quant_conv": [
34
- "optimum",
35
- "NeuronModelVaePostQuantConv"
36
  ]
37
  }
 
1
  {
2
+ "_class_name": "StableDiffusionPipeline",
3
  "_diffusers_version": "0.17.1",
4
  "feature_extractor": [
5
  "transformers",
 
15
  "PNDMScheduler"
16
  ],
17
  "text_encoder": [
18
+ "transformers",
19
+ "CLIPTextModel"
20
  ],
21
  "tokenizer": [
22
  "transformers",
23
  "CLIPTokenizer"
24
  ],
25
  "unet": [
26
+ "diffusers",
27
+ "UNet2DConditionModel"
 
 
 
 
28
  ],
29
+ "vae": [
30
+ "diffusers",
31
+ "AutoencoderKL"
32
  ]
33
  }
text_encoder/model.neuron CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d95bd78fa74d0a6d050db76a4f4b63385b06715999e7d862d68357435bd552fb
3
- size 387704
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8cdfc37c84ab3a8d3bb8321ff39c80cfed900bbcacf6542664a00ee7475a6988
3
+ size 387448
unet/model.neuron CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f7989c4b1645eb51d6ab68b738ca024cc00d6c6f29b12017b02dca3ee1793c38
3
- size 6121820
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bb1ba4d98f033edff245cc793001c71dad40081e6a929e8b628a24eaf1fb7b4f
3
+ size 6122524
vae_decoder/config.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ce99bc11991700ced056faffb4ef430ab41b07f7d841a2902b67696542c3b9e0
3
- size 1093
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a4d48e022043a16a2307fca14a01b49ffc18923f0f546e0e2085623b2f7ab1e1
3
+ size 1098
vae_decoder/model.neuron CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:124093fefa4391c9d30872ea74019ff3ccbe11d403e69a748201211ac3926da5
3
- size 31644
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8aa861a0ff92694257f32b9e7066613b0e5d551c39484bac9d148895af863079
3
+ size 2277468
{vae_post_quant_conv → vae_encoder}/config.json RENAMED
@@ -15,7 +15,7 @@
15
  "in_channels": 3,
16
  "latent_channels": 4,
17
  "layers_per_block": 1,
18
- "model_type": "conv2d",
19
  "neuron": {
20
  "auto_cast": null,
21
  "auto_cast_type": null,
@@ -23,10 +23,10 @@
23
  "compiler_version": "2.6.0.19+3d819e565",
24
  "dynamic_batch_size": false,
25
  "input_names": [
26
- "latent_sample"
27
  ],
28
  "output_names": [
29
- "sample"
30
  ],
31
  "static_batch_size": 1,
32
  "static_height": 64,
 
15
  "in_channels": 3,
16
  "latent_channels": 4,
17
  "layers_per_block": 1,
18
+ "model_type": "vae-encoder",
19
  "neuron": {
20
  "auto_cast": null,
21
  "auto_cast_type": null,
 
23
  "compiler_version": "2.6.0.19+3d819e565",
24
  "dynamic_batch_size": false,
25
  "input_names": [
26
+ "sample"
27
  ],
28
  "output_names": [
29
+ "latent_sample"
30
  ],
31
  "static_batch_size": 1,
32
  "static_height": 64,
{vae_post_quant_conv → vae_encoder}/model.neuron RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:124093fefa4391c9d30872ea74019ff3ccbe11d403e69a748201211ac3926da5
3
- size 31644
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1a1377e2262726b4e1a2c5685422f2cf11bdbb6661a150a0bc5e7533a2f0ecb4
3
+ size 1450076