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:
|
|
@@ -24,6 +24,8 @@ framework:
|
|
| 24 |
action_model_type: DiT-B
|
| 25 |
action_hidden_dim: 768
|
| 26 |
action_dim: 7
|
|
|
|
|
|
|
| 27 |
use_ema: false
|
| 28 |
future_action_window_size: 7
|
| 29 |
past_action_window_size: 0
|
|
|
|
| 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:
|
|
|
|
| 24 |
action_model_type: DiT-B
|
| 25 |
action_hidden_dim: 768
|
| 26 |
action_dim: 7
|
| 27 |
+
input_dim: 2048
|
| 28 |
+
ouptput_dim: 768
|
| 29 |
use_ema: false
|
| 30 |
future_action_window_size: 7
|
| 31 |
past_action_window_size: 0
|