patrickvonplaten commited on
Commit
92d22c6
1 Parent(s): 724cc13
safety_checker/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
- "_commit_hash": "65bb64ec9a0f1c987380ecf78eade396dfa23d14",
3
- "_name_or_path": "/home/patrick/.cache/huggingface/diffusers/models--hf-internal-testing--tiny-stable-diffusion-pipe/snapshots/65bb64ec9a0f1c987380ecf78eade396dfa23d14/safety_checker",
4
  "architectures": [
5
  "StableDiffusionSafetyChecker"
6
  ],
 
1
  {
2
+ "_commit_hash": null,
3
+ "_name_or_path": "./safety_checker",
4
  "architectures": [
5
  "StableDiffusionSafetyChecker"
6
  ],
text_encoder/pytorch_model.no_ema.bin → safety_checker/pytorch_model.fp16.bin RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:80b106be1923d082970d3b6356487dc0b7822f74828abe6a9a5f57c71e12c85f
3
- size 302289
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf963bcd70d3b007da6c367af97e3c5fa388880191e17393f1ef40f9acf90eca
3
+ size 213539
save.py CHANGED
@@ -1,6 +1,10 @@
1
  #!/usr/bin/env python3
2
  from transformers import CLIPTextModel
 
 
3
 
4
- model = CLIPTextModel.from_pretrained("./text_encoder")
5
- model.save_pretrained("./text_encoder", variant="no_ema")
6
- model.save_pretrained("./text_encoder", variant="fp16")
 
 
 
1
  #!/usr/bin/env python3
2
  from transformers import CLIPTextModel
3
+ from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
4
+ from diffusers import AutoencoderKL, UNet2DConditionModel
5
 
6
+ name = "./unet"
7
+ model = UNet2DConditionModel.from_pretrained(name)
8
+
9
+ model.save_pretrained(name, variant="no_ema")
10
+ model.save_pretrained(name, variant="fp16")
unet/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.13.0.dev0",
4
- "_name_or_path": "/home/patrick/.cache/huggingface/diffusers/models--hf-internal-testing--tiny-stable-diffusion-pipe/snapshots/65bb64ec9a0f1c987380ecf78eade396dfa23d14/unet",
5
  "act_fn": "silu",
6
  "attention_head_dim": 8,
7
  "block_out_channels": [
@@ -10,6 +10,8 @@
10
  ],
11
  "center_input_sample": false,
12
  "class_embed_type": null,
 
 
13
  "cross_attention_dim": 32,
14
  "down_block_types": [
15
  "DownBlock2D",
@@ -31,6 +33,9 @@
31
  "out_channels": 4,
32
  "resnet_time_scale_shift": "default",
33
  "sample_size": 32,
 
 
 
34
  "up_block_types": [
35
  "CrossAttnUpBlock2D",
36
  "UpBlock2D"
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.13.0.dev0",
4
+ "_name_or_path": "./unet",
5
  "act_fn": "silu",
6
  "attention_head_dim": 8,
7
  "block_out_channels": [
 
10
  ],
11
  "center_input_sample": false,
12
  "class_embed_type": null,
13
+ "conv_in_kernel": 3,
14
+ "conv_out_kernel": 3,
15
  "cross_attention_dim": 32,
16
  "down_block_types": [
17
  "DownBlock2D",
 
33
  "out_channels": 4,
34
  "resnet_time_scale_shift": "default",
35
  "sample_size": 32,
36
+ "time_cond_proj_dim": null,
37
+ "time_embedding_type": "positional",
38
+ "timestep_post_act": null,
39
  "up_block_types": [
40
  "CrossAttnUpBlock2D",
41
  "UpBlock2D"
unet/diffusion_pytorch_model.fp16.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:56f6ef91ee2b491611f5db5fb39f5e87f9818777d290ee0535f009f665800cec
3
+ size 5831675
unet/diffusion_pytorch_model.no_ema.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0b91407b89ff195ec0ed99d5028fca79c632c490f4c5dd542f177993a861bb6
3
+ size 5832287
vae/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "AutoencoderKL",
3
  "_diffusers_version": "0.13.0.dev0",
4
- "_name_or_path": "/home/patrick/.cache/huggingface/diffusers/models--hf-internal-testing--tiny-stable-diffusion-pipe/snapshots/65bb64ec9a0f1c987380ecf78eade396dfa23d14/vae",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  32,
 
1
  {
2
  "_class_name": "AutoencoderKL",
3
  "_diffusers_version": "0.13.0.dev0",
4
+ "_name_or_path": "./vae",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  32,
vae/diffusion_pytorch_model.fp16.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c6d1d95d5b005bc1f82ec4bc56cceacdf14fca0182f56fa2ff1dcb0594980c5
3
+ size 2681375