Upload 6 files

model_index.json

@@ -1,14 +1,14 @@
 {
   "_class_name": "DiTPipeline",
   "_diffusers_version": "0.12.0.dev0",
-  "dit": [
-    "diffusers",
-    "Transformer2DModel"
-  ],
   "scheduler": [
     "diffusers",
     "DDIMScheduler"
   ],
+  "transformer": [
+    "diffusers",
+    "Transformer2DModel"
+  ],
   "vae": [
     "diffusers",
     "AutoencoderKL"

transformer/config.json

@@ -0,0 +1,23 @@
+{
+  "_class_name": "Transformer2DModel",
+  "_diffusers_version": "0.12.0.dev0",
+  "activation_fn": "gelu-approximate",
+  "attention_bias": true,
+  "attention_head_dim": 72,
+  "cross_attention_dim": null,
+  "dropout": 0.0,
+  "in_channels": 4,
+  "norm_elementwise_affine": false,
+  "norm_num_groups": 32,
+  "norm_type": "ada_norm_zero",
+  "num_attention_heads": 16,
+  "num_embeds_ada_norm": 1000,
+  "num_layers": 28,
+  "num_vector_embeds": null,
+  "only_cross_attention": false,
+  "out_channels": 8,
+  "patch_size": 2,
+  "sample_size": 64,
+  "upcast_attention": false,
+  "use_linear_projection": false
+}

transformer/diffusion_pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b00d4be3612dd89bda715ffdcd9efe1635605f52d292ddd6f600dd0c8097517a
+size 2999533581