Upload PixArtSigmaPipeline

transformer/config.json

@@ -5,7 +5,7 @@
   "attention_bias": true,
   "attention_head_dim": 72,
   "attention_type": "default",
-  "caption_channels": 4096,
+  "caption_channels": null,
   "cross_attention_dim": 1152,
   "dropout": 0.0,
   "in_channels": 16,

transformer/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c8ad476dc9f0fd5e8080130e9f734d99d5d259c845fdf1cbfdc41f02a7cb414b
-size 2444156432
+oid sha256:6a08a9aad6fa1aeb417d308863c433ea478677f73fa5aec5c5d965517f081123
+size 2419963984

vae/config.json

@@ -1,12 +1,12 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.30.2",
-  "_name_or_path": "AuraDiffusion/16ch-vae",
+  "_name_or_path": "ostris/vae-kl-f8-d16",
   "act_fn": "silu",
   "block_out_channels": [
+    128,
     128,
     256,
-    512,
     512
   ],
   "down_block_types": [
@@ -15,18 +15,18 @@
     "DownEncoderBlock2D",
     "DownEncoderBlock2D"
   ],
-  "force_upcast": false,
+  "force_upcast": true,
   "in_channels": 3,
   "latent_channels": 16,
   "latents_mean": null,
   "latents_std": null,
   "layers_per_block": 2,
-  "mid_block_add_attention": false,
+  "mid_block_add_attention": true,
   "norm_num_groups": 32,
   "out_channels": 3,
-  "sample_size": 1024,
-  "scaling_factor": 0.18215,
-  "shift_factor": 0,
+  "sample_size": 512,
+  "scaling_factor": 0.2614,
+  "shift_factor": null,
   "up_block_types": [
     "UpDecoderBlock2D",
     "UpDecoderBlock2D",

vae/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ead97b12214d10d347a57425f2387242710c8e101e09466bcf91bc0f28d65e9c
-size 326896452
+oid sha256:b6468f242218bc72981284b6fda570cc7c07ce043e84979cc1bb7d99d39a64a2
+size 229094420