Update config.yaml
Browse files- 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: /
|
| 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
|