Hugging Face's logo

optimum
/
tiny-stable-diffusion-neuronx

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

update

Browse files
Files changed (3) hide show
  1. model_index.json +12 -8
  2. vae_decoder/config.json +2 -2
  3. vae_decoder/model.neuron +2 -2
model_index.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_class_name": "StableDiffusionPipeline",
3
  "_diffusers_version": "0.17.1",
4
  "feature_extractor": [
5
  "transformers",
@@ -15,19 +15,23 @@
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
  }
 
1
  {
2
+ "_class_name": "NeuronStableDiffusionPipeline",
3
  "_diffusers_version": "0.17.1",
4
  "feature_extractor": [
5
  "transformers",
 
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
  }
vae_decoder/config.json CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a4d48e022043a16a2307fca14a01b49ffc18923f0f546e0e2085623b2f7ab1e1
3
- size 1098
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ce99bc11991700ced056faffb4ef430ab41b07f7d841a2902b67696542c3b9e0
3
+ size 1093
vae_decoder/model.neuron CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:83893ff1c38cbd94a70d61015d4dda7d83e428ff5b214ed007ef759fdbcd1386
3
- size 2314780
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:124093fefa4391c9d30872ea74019ff3ccbe11d403e69a748201211ac3926da5
3
+ size 31644