Update config.yaml
Browse files- config.yaml +2 -0
config.yaml
CHANGED
|
@@ -20,6 +20,8 @@ framework:
|
|
| 20 |
qformer_start_layer: 36
|
| 21 |
num_query_tokens: 64
|
| 22 |
grad_scale: 0.5
|
|
|
|
|
|
|
| 23 |
action_model:
|
| 24 |
action_model_type: DiT-B
|
| 25 |
action_hidden_dim: 768
|
|
|
|
| 20 |
qformer_start_layer: 36
|
| 21 |
num_query_tokens: 64
|
| 22 |
grad_scale: 0.5
|
| 23 |
+
input_dim: 2048
|
| 24 |
+
ouptput_dim: 768
|
| 25 |
action_model:
|
| 26 |
action_model_type: DiT-B
|
| 27 |
action_hidden_dim: 768
|