Upload 6 files

dit/config.json

@@ -1,14 +1,23 @@
 {
-  "_class_name": "DiT",
+  "_class_name": "Transformer2DModel",
   "_diffusers_version": "0.12.0.dev0",
-  "class_dropout_prob": 0.1,
-  "depth": 28,
-  "hidden_size": 1152,
+  "activation_fn": "gelu-approximate",
+  "attention_bias": true,
+  "attention_head_dim": 72,
+  "cross_attention_dim": null,
+  "dropout": 0.0,
   "in_channels": 4,
-  "input_size": 64,
-  "learn_sigma": true,
-  "mlp_ratio": 4,
-  "num_classes": 1000,
-  "num_heads": 16,
-  "patch_size": 2
+  "norm_elementwise_affine": false,
+  "norm_num_groups": 32,
+  "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_ada_layer_norm_zero": true,
+  "use_linear_projection": false
 }

dit/diffusion_pytorch_model.bin

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

model_index.json

@@ -3,7 +3,7 @@
   "_diffusers_version": "0.12.0.dev0",
   "dit": [
     "diffusers",
-    "DiT"
+    "Transformer2DModel"
   ],
   "scheduler": [
     "diffusers",