fp16 weights

safety_checker/config.json

@@ -1,6 +1,6 @@
 {
-  "_commit_hash": "4bb648a606ef040e7685bde262611766a5fdd67b",
-  "_name_or_path": "CompVis/stable-diffusion-safety-checker",
+  "_commit_hash": null,
+  "_name_or_path": "./AltDiffusion/safety_checker",
   "architectures": [
     "StableDiffusionSafetyChecker"
   ],
@@ -79,7 +79,7 @@
     "top_p": 1.0,
     "torch_dtype": null,
     "torchscript": false,
-    "transformers_version": "4.24.0.dev0",
+    "transformers_version": "4.23.1",
     "typical_p": 1.0,
     "use_bfloat16": false,
     "vocab_size": 49408
@@ -90,7 +90,7 @@
     "num_attention_heads": 12,
     "num_hidden_layers": 12
   },
-  "torch_dtype": "float32",
+  "torch_dtype": "float16",
   "transformers_version": null,
   "vision_config": {
     "_name_or_path": "",
@@ -165,7 +165,7 @@
     "top_p": 1.0,
     "torch_dtype": null,
     "torchscript": false,
-    "transformers_version": "4.24.0.dev0",
+    "transformers_version": "4.23.1",
     "typical_p": 1.0,
     "use_bfloat16": false
   },

safety_checker/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7cebe6408f67beaf5e2272c4a90244472c7c54bba28dd27a21ec8b4b98f5bcd9
-size 1216071332
+oid sha256:70b4ace4d3096de0a0c6d3574377bbaa17c6338daa9c728ea32f9cfd80a8907c
+size 608101569

scheduler/scheduler_config.json

@@ -1,13 +1,13 @@
 {
   "_class_name": "PNDMScheduler",
-  "_diffusers_version": "0.6.0",
+  "_diffusers_version": "0.8.0.dev0",
   "beta_end": 0.012,
   "beta_schedule": "scaled_linear",
   "beta_start": 0.00085,
+  "clip_sample": false,
   "num_train_timesteps": 1000,
   "set_alpha_to_one": false,
   "skip_prk_steps": true,
   "steps_offset": 1,
-  "trained_betas": null,
-  "clip_sample": false
+  "trained_betas": null
 }

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "../test/text_encoder",
+  "_name_or_path": "./AltDiffusion/text_encoder",
   "architectures": [
     "RobertaSeriesModelWithTransformation"
   ],
@@ -23,8 +23,8 @@
   "pooler_fn": "cls",
   "position_embedding_type": "absolute",
   "project_dim": 768,
-  "torch_dtype": "float32",
-  "transformers_version": "4.24.0.dev0",
+  "torch_dtype": "float16",
+  "transformers_version": "4.23.1",
   "type_vocab_size": 1,
   "use_attention_mask": true,
   "use_cache": true,

text_encoder/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6f7e5b6bef6737f75d01705be1d59295f5c9d84c10b9715a1610160fb646245d
-size 2242851593
+oid sha256:ca6def2aab8425a75e78318298e715bcd6ba8366646b0778bd1a1ddc9661ca4a
+size 1121489845

tokenizer/tokenizer_config.json

@@ -11,7 +11,7 @@
     "single_word": false
   },
   "model_max_length": 77,
-  "name_or_path": "../test/tokenizer",
+  "name_or_path": "./AltDiffusion/tokenizer",
   "pad_token": "<pad>",
   "sep_token": "</s>",
   "sp_model_kwargs": {},

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.8.0.dev0",
-  "_name_or_path": "../test/unet",
+  "_name_or_path": "./AltDiffusion/unet",
   "act_fn": "silu",
   "attention_head_dim": 8,
   "block_out_channels": [

unet/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3ff603f7757f3efde7e92bac8d667e40c4c742634bee84bf906f3a2ff2a99b46
-size 3438370637
+oid sha256:bac854229d5b1b459eb5dfd0ca88827d943abfa82c095aec4ba232f94d378db4
+size 1719324453

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.8.0.dev0",
-  "_name_or_path": "../test/vae",
+  "_name_or_path": "./AltDiffusion/vae",
   "act_fn": "silu",
   "block_out_channels": [
     128,

vae/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c8b4d738e8f8ecefacb7ace7e5bc25a4a7fec79606197183892e46cbb304a227
-size 334713511
+oid sha256:f6e817ec7c5c6641c0000a1a7fc7893f6625251131f33f7bb034c55e5d949455
+size 167404145