patrickvonplaten
commited on
Commit
•
92d22c6
1
Parent(s):
724cc13
up
Browse files- safety_checker/config.json +2 -2
- text_encoder/pytorch_model.no_ema.bin → safety_checker/pytorch_model.fp16.bin +2 -2
- save.py +7 -3
- unet/config.json +6 -1
- unet/diffusion_pytorch_model.fp16.bin +3 -0
- unet/diffusion_pytorch_model.no_ema.bin +3 -0
- vae/config.json +1 -1
- vae/diffusion_pytorch_model.fp16.bin +3 -0
safety_checker/config.json
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
{
|
2 |
-
"_commit_hash":
|
3 |
-
"_name_or_path": "
|
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:
|
3 |
-
size
|
|
|
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 |
-
|
5 |
-
model.
|
6 |
-
|
|
|
|
|
|
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": "
|
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": "
|
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
|