Update model config

README.md

@@ -22,5 +22,5 @@ The model is converted into `.pte` format for deployment on mobile devices.
 
 ### Original Model Sources
 
-- **Github (original):** [BK-SDM](https://github.com/Nota-NetsPresso/BK-SDM)
-- **Paper:** [On Architectural Compression of Text-to-Image Diffusion Models (arXiv)](https://arxiv.org/abs/2305.15798)
+- **Github:** [BK-SDM](https://github.com/Nota-NetsPresso/BK-SDM)
+- **Paper:** [On Architectural Compression of Text-to-Image Diffusion Models (arXiv)](https://arxiv.org/abs/2305.15798)

model_index.json

@@ -0,0 +1,33 @@
+{
+  "_class_name": "StableDiffusionPipeline",
+  "_diffusers_version": "0.20.2",
+  "feature_extractor": [
+    "transformers",
+    "CLIPImageProcessor"
+  ],
+  "safety_checker": [
+    "stable_diffusion",
+    "StableDiffusionSafetyChecker"
+  ],
+  "scheduler": [
+    "diffusers",
+    "PNDMScheduler"
+  ],
+  "text_encoder": [
+    "transformers",
+    "CLIPTextModel"
+  ],
+  "tokenizer": [
+    "transformers",
+    "CLIPTokenizer"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DConditionModel"
+  ],
+  "vae": [
+    "diffusers",
+    "AutoencoderKL"
+  ]
+}
+

text_encoder_model.pte

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:2563fa08c678182ee81ef8f25a625ed4c1d3896ba17e430b6ee788ba9c868a47
-size 492446592

unet_model.pte

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:8328c045e470813fdd043cbc53bd4fd3832b544e5d3819011010a55029eb92cd
-size 1293966864

vae_model.pte

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:1749cfa4ddf476cdb5ee83eaf602495db56fb80b442919d6812d2fa5f0a208a2
-size 198074508