flux-3-action-droid / config.json
Nicewood's picture
Add FLUX 3 Action droid package with shared encoders
ea77cad verified
Raw History Blame
2.1 kB
{
"type": "flux3",
"n_obs_steps": 1,
"input_features": {
"observation.images.wrist_image_left": {
"type": "VISUAL",
"shape": [
3,
360,
640
]
},
"observation.images.exterior_image_1_left": {
"type": "VISUAL",
"shape": [
3,
360,
640
]
},
"observation.images.exterior_image_2_left": {
"type": "VISUAL",
"shape": [
3,
360,
640
]
},
"observation.state": {
"type": "STATE",
"shape": [
8
]
}
},
"output_features": {
"action": {
"type": "ACTION",
"shape": [
8
]
}
},
"device": "cuda",
"use_amp": false,
"use_peft": false,
"conditioning": "frame",
"packer": null,
"condition_on_past_actions": false,
"action_representation": "absolute",
"delta_absolute_dims": [],
"text_fixed_length": null,
"video_position_fps": null,
"history_snapshots": 1,
"chunk_size": 32,
"n_action_steps": 32,
"fps": 15.0,
"camera_layout": "droid",
"camera_keys": [
"observation.images.wrist_image_left",
"observation.images.exterior_image_1_left",
"observation.images.exterior_image_2_left"
],
"canvas_hw": [
544,
736
],
"action_modality": "action_prediction_droid",
"action_scale": 2.0,
"gripper_flip_dims": [
-1
],
"use_relative_actions": false,
"relative_exclude_joints": [
"gripper"
],
"action_feature_names": null,
"normalization_mapping": {
"VISUAL": "IDENTITY",
"STATE": "IDENTITY",
"ACTION": "IDENTITY"
},
"video_vae_id": "black-forest-labs/flux-3-action-base:video_vae.safetensors@62878e2925e59b7a89ec14463ce89932624c490d",
"text_encoder_id": "black-forest-labs/flux-3-action-base:text_encoder@62878e2925e59b7a89ec14463ce89932624c490d",
"dit_config": null,
"attn_mode": "torch",
"dtype": "bfloat16",
"compile_model": false,
"sampler": "cosmos_unipc",
"num_inference_steps": 4,
"guidance_scale": 4.0,
"guidance_scale_action": 1.0,
"sampler_shift": 5.0,
"inference_seed": 0
}