up

safety_checker/config.json

@@ -1,6 +1,6 @@
 {
-  "_commit_hash": "65bb64ec9a0f1c987380ecf78eade396dfa23d14",
-  "_name_or_path": "/home/patrick/.cache/huggingface/diffusers/models--hf-internal-testing--tiny-stable-diffusion-pipe/snapshots/65bb64ec9a0f1c987380ecf78eade396dfa23d14/safety_checker",
+  "_commit_hash": null,
+  "_name_or_path": "./safety_checker",
   "architectures": [
     "StableDiffusionSafetyChecker"
   ],

text_encoder/pytorch_model.no_ema.bin → safety_checker/pytorch_model.fp16.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:80b106be1923d082970d3b6356487dc0b7822f74828abe6a9a5f57c71e12c85f
-size 302289
+oid sha256:bf963bcd70d3b007da6c367af97e3c5fa388880191e17393f1ef40f9acf90eca
+size 213539

save.py

@@ -1,6 +1,10 @@
 #!/usr/bin/env python3
 from transformers import CLIPTextModel
+from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
+from diffusers import AutoencoderKL, UNet2DConditionModel
 
-model = CLIPTextModel.from_pretrained("./text_encoder")
-model.save_pretrained("./text_encoder", variant="no_ema")
-model.save_pretrained("./text_encoder", variant="fp16")
+name = "./unet"
+model = UNet2DConditionModel.from_pretrained(name)
+
+model.save_pretrained(name, variant="no_ema")
+model.save_pretrained(name, variant="fp16")

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.13.0.dev0",
-  "_name_or_path": "/home/patrick/.cache/huggingface/diffusers/models--hf-internal-testing--tiny-stable-diffusion-pipe/snapshots/65bb64ec9a0f1c987380ecf78eade396dfa23d14/unet",
+  "_name_or_path": "./unet",
   "act_fn": "silu",
   "attention_head_dim": 8,
   "block_out_channels": [
@@ -10,6 +10,8 @@
   ],
   "center_input_sample": false,
   "class_embed_type": null,
+  "conv_in_kernel": 3,
+  "conv_out_kernel": 3,
   "cross_attention_dim": 32,
   "down_block_types": [
     "DownBlock2D",
@@ -31,6 +33,9 @@
   "out_channels": 4,
   "resnet_time_scale_shift": "default",
   "sample_size": 32,
+  "time_cond_proj_dim": null,
+  "time_embedding_type": "positional",
+  "timestep_post_act": null,
   "up_block_types": [
     "CrossAttnUpBlock2D",
     "UpBlock2D"

unet/diffusion_pytorch_model.fp16.bin

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

unet/diffusion_pytorch_model.no_ema.bin

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

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.13.0.dev0",
-  "_name_or_path": "/home/patrick/.cache/huggingface/diffusers/models--hf-internal-testing--tiny-stable-diffusion-pipe/snapshots/65bb64ec9a0f1c987380ecf78eade396dfa23d14/vae",
+  "_name_or_path": "./vae",
   "act_fn": "silu",
   "block_out_channels": [
     32,

vae/diffusion_pytorch_model.fp16.bin

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