Upload 18 files

text_encoder/config.json

@@ -18,7 +18,7 @@
   "num_hidden_layers": 12,
   "pad_token_id": 1,
   "projection_dim": 768,
-  "torch_dtype": "float16",
+  "torch_dtype": "bfloat16",
   "transformers_version": "4.43.0.dev0",
   "vocab_size": 49408
 }

text_encoder/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5e4717fdea5fbd8f6940d289364eba69a2dbf8a037856b0f576c621e64365e25
-size 246144152
+oid sha256:a34faf6bcff6ed89e66d5071c2d034902d898d15156ce229caa28a21a8e4c565
+size 246144352

text_encoder_2/config.json

@@ -18,7 +18,7 @@
   "num_hidden_layers": 32,
   "pad_token_id": 1,
   "projection_dim": 1280,
-  "torch_dtype": "float16",
+  "torch_dtype": "bfloat16",
   "transformers_version": "4.43.0.dev0",
   "vocab_size": 49408
 }

text_encoder_2/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:621bb27d7b6a16035a7241e1b34752c57d4b3dc96428f2cd2a5245e0aaeaa6c0
-size 1389382176
+oid sha256:6c3edbf47b12ed6a462b54ec11f67784015fbf413ab6569f39b791e0eb8d23b4
+size 1389382688

unet/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cb111ced1ebdcfcf7d8da17b43e35cbf2db2ab775ffa4067c4ddeca0e9ac6ede
-size 5135149760
+oid sha256:4ef290b82476746b602d58286669c01583f7d17cf382dccd92b2b429d45626bc
+size 5135151440

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.30.0.dev0",
-  "_name_or_path": "madebyollin/sdxl-vae-fp16-fix",
+  "_name_or_path": "../sdxl-vae/",
   "act_fn": "silu",
   "block_out_channels": [
     128,
@@ -15,7 +15,7 @@
     "DownEncoderBlock2D",
     "DownEncoderBlock2D"
   ],
-  "force_upcast": false,
+  "force_upcast": true,
   "in_channels": 3,
   "latent_channels": 4,
   "latents_mean": null,
@@ -23,7 +23,7 @@
   "layers_per_block": 2,
   "norm_num_groups": 32,
   "out_channels": 3,
-  "sample_size": 512,
+  "sample_size": 1024,
   "scaling_factor": 0.13025,
   "shift_factor": null,
   "up_block_types": [

vae/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6353737672c94b96174cb590f711eac6edf2fcce5b6e91aa9d73c5adc589ee48
-size 167335342
+oid sha256:9974e9bd90b834b8cb74fa53a63a64643fbeb02438301ac1adde0c042f70fd24
+size 167335590