optimum
/

tiny-stable-diffusion-neuronx

Model card Files Files and versions Community

Jingya HF staff commited on Jul 11, 2023

Commit

0f5606d

•

1 Parent(s): 12cb6fa

update

Files changed (3) hide show

model_index.json +12 -8
vae_decoder/config.json +2 -2
vae_decoder/model.neuron +2 -2

model_index.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "_class_name": "StableDiffusionPipeline",
   "_diffusers_version": "0.17.1",
   "feature_extractor": [
     "transformers",
@@ -15,19 +15,23 @@
     "PNDMScheduler"
   ],
   "text_encoder": [
-    "transformers",
-    "CLIPTextModel"
   ],
   "tokenizer": [
     "transformers",
     "CLIPTokenizer"
   ],
   "unet": [
-    "diffusers",
-    "UNet2DConditionModel"
   ],
-  "vae": [
-    "diffusers",
-    "AutoencoderKL"
   ]
 }

 {
+  "_class_name": "NeuronStableDiffusionPipeline",
   "_diffusers_version": "0.17.1",
   "feature_extractor": [
     "transformers",
     "PNDMScheduler"
   ],
   "text_encoder": [
+    "optimum",
+    "NeuronModelTextEncoder"
   ],
   "tokenizer": [
     "transformers",
     "CLIPTokenizer"
   ],
   "unet": [
+    "optimum",
+    "NeuronModelUnet"
   ],
+  "vae_decoder": [
+    "optimum",
+    "NeuronModelVaeDecoder"
+  ],
+  "vae_post_quant_conv": [
+    "optimum",
+    "NeuronModelVaePostQuantConv"
   ]
 }

vae_decoder/config.json CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a4d48e022043a16a2307fca14a01b49ffc18923f0f546e0e2085623b2f7ab1e1
-size 1098

 version https://git-lfs.github.com/spec/v1
+oid sha256:ce99bc11991700ced056faffb4ef430ab41b07f7d841a2902b67696542c3b9e0
+size 1093

vae_decoder/model.neuron CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:83893ff1c38cbd94a70d61015d4dda7d83e428ff5b214ed007ef759fdbcd1386
-size 2314780

 version https://git-lfs.github.com/spec/v1
+oid sha256:124093fefa4391c9d30872ea74019ff3ccbe11d403e69a748201211ac3926da5
+size 31644