diff --git a/orchestrator/ui/src/components/InferenceModelSelector.js b/orchestrator/ui/src/components/InferenceModelSelector.js index bf5487d..03360fb 100644 --- a/orchestrator/ui/src/components/InferenceModelSelector.js +++ b/orchestrator/ui/src/components/InferenceModelSelector.js @@ -36,6 +36,7 @@ const MODEL_GROUPS = [ { value: 'lerobot:pi0', label: 'Pi0', serviceType: 'lerobot', policyType: 'pi0' }, { value: 'lerobot:pi05', label: 'Pi0.5', serviceType: 'lerobot', policyType: 'pi05' }, { value: 'lerobot:diffusion', label: 'Diffusion', serviceType: 'lerobot', policyType: 'diffusion' }, + { value: 'lerobot:vqbet', label: 'VQ-BeT', serviceType: 'lerobot', policyType: 'vqbet' }, ], }, {