update

README.md

@@ -13,4 +13,4 @@ stable_diffusion.save_pretrained(save_directory)
 
 prompt = "sailing ship in storm by Leonardo da Vinci"
 image = stable_diffusion(prompt).images[0]
-```
+```

model_index.json

@@ -1,5 +1,5 @@
 {
-  "_class_name": "NeuronStableDiffusionPipeline",
+  "_class_name": "StableDiffusionPipeline",
   "_diffusers_version": "0.17.1",
   "feature_extractor": [
     "transformers",
@@ -15,23 +15,19 @@
     "PNDMScheduler"
   ],
   "text_encoder": [
-    "optimum",
-    "NeuronModelTextEncoder"
+    "transformers",
+    "CLIPTextModel"
   ],
   "tokenizer": [
     "transformers",
     "CLIPTokenizer"
   ],
   "unet": [
-    "optimum",
-    "NeuronModelUnet"
-  ],
-  "vae_decoder": [
-    "optimum",
-    "NeuronModelVaeDecoder"
+    "diffusers",
+    "UNet2DConditionModel"
   ],
-  "vae_post_quant_conv": [
-    "optimum",
-    "NeuronModelVaePostQuantConv"
+  "vae": [
+    "diffusers",
+    "AutoencoderKL"
   ]
 }

text_encoder/model.neuron

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d95bd78fa74d0a6d050db76a4f4b63385b06715999e7d862d68357435bd552fb
-size 387704
+oid sha256:8cdfc37c84ab3a8d3bb8321ff39c80cfed900bbcacf6542664a00ee7475a6988
+size 387448

unet/model.neuron

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f7989c4b1645eb51d6ab68b738ca024cc00d6c6f29b12017b02dca3ee1793c38
-size 6121820
+oid sha256:bb1ba4d98f033edff245cc793001c71dad40081e6a929e8b628a24eaf1fb7b4f
+size 6122524

vae_decoder/config.json

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

vae_decoder/model.neuron

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:124093fefa4391c9d30872ea74019ff3ccbe11d403e69a748201211ac3926da5
-size 31644
+oid sha256:8aa861a0ff92694257f32b9e7066613b0e5d551c39484bac9d148895af863079
+size 2277468

{vae_post_quant_conv → vae_encoder}/config.json

@@ -15,7 +15,7 @@
   "in_channels": 3,
   "latent_channels": 4,
   "layers_per_block": 1,
-  "model_type": "conv2d",
+  "model_type": "vae-encoder",
   "neuron": {
     "auto_cast": null,
     "auto_cast_type": null,
@@ -23,10 +23,10 @@
     "compiler_version": "2.6.0.19+3d819e565",
     "dynamic_batch_size": false,
     "input_names": [
-      "latent_sample"
+      "sample"
     ],
     "output_names": [
-      "sample"
+      "latent_sample"
     ],
     "static_batch_size": 1,
     "static_height": 64,

{vae_post_quant_conv → vae_encoder}/model.neuron

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:124093fefa4391c9d30872ea74019ff3ccbe11d403e69a748201211ac3926da5
-size 31644
+oid sha256:1a1377e2262726b4e1a2c5685422f2cf11bdbb6661a150a0bc5e7533a2f0ecb4
+size 1450076