Upload folder using huggingface_hub

config.json

@@ -12,6 +12,8 @@
   "model_type": "dit",
   "num_attention_heads": 16,
   "num_hidden_layers": 24,
+  "p_uniform": 0.1,
+  "t_eps": 0.0001,
   "timestep_cond_dim": 128,
   "torch_dtype": "float32",
   "transformers_version": "4.49.0",

configuration_dit.py

@@ -13,6 +13,8 @@ class DITConfig(PretrainedConfig):
         num_hidden_layers: int = 12,
         num_attention_heads: int = 12,
         attention_dropout: float = 0.0,
+        p_uniform: float = 0.0,
+        t_eps: float = 1e-4,
         **kwargs
     ):
         super().__init__(**kwargs)
@@ -23,3 +25,5 @@ class DITConfig(PretrainedConfig):
         self.num_hidden_layers = num_hidden_layers
         self.num_attention_heads = num_attention_heads
         self.attention_dropout = attention_dropout
+        self.p_uniform = p_uniform
+        self.t_eps = t_eps