[init] initial commit

This commit is contained in:
fenghao.2019
2025-05-26 23:20:51 +08:00
commit 49f51871c6
31 changed files with 2757 additions and 0 deletions
+17
View File
@@ -0,0 +1,17 @@
model:
model_name_or_path: "./checkpoints/dolphin_model.bin"
tokenizer_path: "./checkpoints/dolphin_tokenizer.json"
extra_answer_tokens: True # add <Answer/> token
max_length: 4096
decoder_layer: 10
max_position_embeddings: 4096
hidden_dimension: 1024
swin_args:
name: 'swin'
img_size: [896, 896]
patch_size: 4
embed_dim: 128
align_long_axis: False
window_size: 7
encoder_layer: [2, 2, 14, 2]
num_heads: [4, 8, 16, 32]