Jinhuiye commited on
Commit
785eb33
·
verified ·
1 Parent(s): 42d5651

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +3 -1
config.yaml CHANGED
@@ -10,7 +10,7 @@ is_debug: false
10
  framework:
11
  framework_py: DinoQFormerACT
12
  qwenvl:
13
- base_vlm: /mnt/phwfile/efm_t/zhuyangkun_tmp_need_del/exp/exp_08_09/manip_sys2_qwen25_3b_onevision_molmo_a0all_refsp20/checkpoint-20000
14
  attn_implementation: flash_attention_2
15
  vl_hidden_dim: 2048
16
  dino:
@@ -19,6 +19,8 @@ framework:
19
  qformer_end_layer: 37
20
  qformer_start_layer: 36
21
  num_query_tokens: 64
 
 
22
  grad_scale: 0.5
23
  action_model:
24
  action_model_type: DiT-B
 
10
  framework:
11
  framework_py: DinoQFormerACT
12
  qwenvl:
13
+ base_vlm: Qwen/Qwen2.5-VL-3B-Instruct
14
  attn_implementation: flash_attention_2
15
  vl_hidden_dim: 2048
16
  dino:
 
19
  qformer_end_layer: 37
20
  qformer_start_layer: 36
21
  num_query_tokens: 64
22
+ input_dim: 2048
23
+ ouptput_dim: 768
24
  grad_scale: 0.5
25
  action_model:
26
  action_model_type: DiT-B