Upload StableNormalPipeline

README.md

@@ -1,3 +1,4 @@
 ---
 license: apache-2.0
+library_name: diffusers
 ---

controlnet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "ControlNetVAEModel",
   "_diffusers_version": "0.28.0",
-  "_name_or_path": "./weights/sgdrn_controlnet/",
+  "_name_or_path": "weights/stable-normal-v0-1/controlnet",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,

controlnet/diffusion_pytorch_model.fp16.safetensors

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

dino_controlnet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "DINOControlNetVAEModel",
   "_diffusers_version": "0.28.0",
-  "_name_or_path": "./weights/sgdrn_dinocontrolnet/",
+  "_name_or_path": "weights/stable-normal-v0-1/dino_controlnet",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,

dino_controlnet/diffusion_pytorch_model.fp16.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:97a89293403c71b33376a941b98fdb8be5c3cb42ad3b6ce0f34c0fb74a3b0a42
+size 749704376

model_index.json

@@ -1,13 +1,13 @@
 {
   "_class_name": "StableNormalPipeline",
   "_diffusers_version": "0.28.0",
-  "_name_or_path": "./weights/sgdrn_sd/",
+  "_name_or_path": "weights/stable-normal-v0-1",
   "controlnet": [
-    "controlnetvae",
+    "diffusers_modules.local.controlnetvae",
     "ControlNetVAEModel"
   ],
   "dino_controlnet": [
-    "dino_controlnetvae",
+    "diffusers_modules.local.dino_controlnetvae",
     "DINOControlNetVAEModel"
   ],
   "feature_extractor": [
@@ -23,6 +23,10 @@
     null,
     null
   ],
+  "scheduler": [
+    "stablenormal.scheduler.heuristics_ddimsampler",
+    "HEURI_DDIMScheduler"
+  ],
   "text_encoder": [
     "transformers",
     "CLIPTextModel"

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "./weights/sgdrn_sd/text_encoder",
+  "_name_or_path": "weights/stable-normal-v0-1/text_encoder",
   "architectures": [
     "CLIPTextModel"
   ],
@@ -19,7 +19,7 @@
   "num_hidden_layers": 23,
   "pad_token_id": 1,
   "projection_dim": 512,
-  "torch_dtype": "float32",
-  "transformers_version": "4.39.1",
+  "torch_dtype": "float16",
+  "transformers_version": "4.41.1",
   "vocab_size": 49408
 }

text_encoder/model.fp16.safetensors

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

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.28.0",
-  "_name_or_path": "./weights/sgdrn_sd/unet",
+  "_name_or_path": "weights/stable-normal-v0-1/unet",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,

unet/diffusion_pytorch_model.fp16.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6138d8250972d1e5b8a13ea8490bba91400ed13c765f41ba07bd639f496a9787
+size 1731904736

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.28.0",
-  "_name_or_path": "./weights/sgdrn_sd/vae",
+  "_name_or_path": "weights/stable-normal-v0-1/vae",
   "act_fn": "silu",
   "block_out_channels": [
     128,

vae/diffusion_pytorch_model.fp16.safetensors

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