From d474382a5c2a751b93adda22a04a560f919fe7b7 Mon Sep 17 00:00:00 2001
From: Ajay Jain <ajayj@users.noreply.huggingface.co>
Date: Tue, 22 Oct 2024 22:35:25 +0000
Subject: [PATCH] Remove unnecessary arguments from config

---
 dit-config.yaml | 11 +----------
 1 file changed, 1 insertion(+), 10 deletions(-)

diff --git a/dit-config.yaml b/dit-config.yaml
index bc74b13..73e1512 100644
--- a/dit-config.yaml
+++ b/dit-config.yaml
@@ -1,4 +1,3 @@
-_target_: ignore_this_field
 depth: 48
 patch_size: 2
 num_heads: 24
@@ -6,21 +5,13 @@ hidden_size_x: 3072
 hidden_size_y: 1536
 mlp_ratio_x: 4.0
 mlp_ratio_y: 4.0
-learn_sigma: false
 in_channels: 12
-clip_feat_dim: 2048
 qk_norm: true
 qkv_bias: false
 out_bias: true
-attn_drop: 0.0
 patch_embed_bias: true
-posenc_preserve_area: true
 timestep_mlp_bias: true
-pooled_caption_mlp_bias: true
-attend_to_padding: false
 timestep_scale: 1000.0
-use_t5: true
 t5_feat_dim: 4096
 t5_token_length: 256
-rope_theta: 10000.0
-use_transformer_engine: true
+rope_theta: 10000.0
\ No newline at end of file