import { z } from "zod";
export declare const vlaConfigSchema: z.ZodObject<{
    backend: z.ZodOptional<z.ZodEnum<{
        cpu: "cpu";
        auto: "auto";
    }>>;
    verbosity: z.ZodOptional<z.ZodNumber>;
}, z.core.$strip>;
export type VlaConfig = z.input<typeof vlaConfigSchema>;
export declare const vlaHparamsSchema: z.ZodObject<{
    chunkSize: z.ZodNumber;
    actionDim: z.ZodNumber;
    maxActionDim: z.ZodNumber;
    maxStateDim: z.ZodNumber;
    tokenizerMaxLength: z.ZodNumber;
    visionImageSize: z.ZodNumber;
    numCameras: z.ZodOptional<z.ZodNumber>;
    stateInputMode: z.ZodOptional<z.ZodEnum<{
        discrete: "discrete";
        continuous: "continuous";
    }>>;
}, z.core.$strip>;
export type VlaHparams = z.infer<typeof vlaHparamsSchema>;
export declare const vlaStatsSchema: z.ZodObject<{
    vision_ms: z.ZodOptional<z.ZodNumber>;
    prefill_compute_ms: z.ZodOptional<z.ZodNumber>;
    prefill_total_ms: z.ZodOptional<z.ZodNumber>;
    smollm2_compute_ms: z.ZodOptional<z.ZodNumber>;
    smollm2_total_ms: z.ZodOptional<z.ZodNumber>;
    ode_ms: z.ZodOptional<z.ZodNumber>;
    total_ms: z.ZodOptional<z.ZodNumber>;
    backendDevice: z.ZodOptional<z.ZodNumber>;
}, z.core.$strip>;
export type VlaStats = z.infer<typeof vlaStatsSchema>;
export declare const vlaRunRequestSchema: z.ZodObject<{
    type: z.ZodLiteral<"vlaRun">;
    modelId: z.ZodString;
    images: z.ZodArray<z.ZodString>;
    imgWidth: z.ZodNumber;
    imgHeight: z.ZodNumber;
    state: z.ZodString;
    tokens: z.ZodString;
    mask: z.ZodString;
    noise: z.ZodOptional<z.ZodString>;
}, z.core.$strip>;
export type VlaRunRequest = z.input<typeof vlaRunRequestSchema>;
export declare const vlaRunResponseSchema: z.ZodObject<{
    actions: z.ZodString;
    actionDim: z.ZodNumber;
    chunkSize: z.ZodNumber;
    stats: z.ZodOptional<z.ZodObject<{
        vision_ms: z.ZodOptional<z.ZodNumber>;
        prefill_compute_ms: z.ZodOptional<z.ZodNumber>;
        prefill_total_ms: z.ZodOptional<z.ZodNumber>;
        smollm2_compute_ms: z.ZodOptional<z.ZodNumber>;
        smollm2_total_ms: z.ZodOptional<z.ZodNumber>;
        ode_ms: z.ZodOptional<z.ZodNumber>;
        total_ms: z.ZodOptional<z.ZodNumber>;
        backendDevice: z.ZodOptional<z.ZodNumber>;
    }, z.core.$strip>>;
}, z.core.$strip>;
export type VlaRunResponse = z.infer<typeof vlaRunResponseSchema>;
export declare const vlaHparamsRequestSchema: z.ZodObject<{
    type: z.ZodLiteral<"vlaHparams">;
    modelId: z.ZodString;
}, z.core.$strip>;
export type VlaHparamsRequest = z.input<typeof vlaHparamsRequestSchema>;
export declare const vlaHparamsResponseSchema: z.ZodObject<{
    hparams: z.ZodObject<{
        chunkSize: z.ZodNumber;
        actionDim: z.ZodNumber;
        maxActionDim: z.ZodNumber;
        maxStateDim: z.ZodNumber;
        tokenizerMaxLength: z.ZodNumber;
        visionImageSize: z.ZodNumber;
        numCameras: z.ZodOptional<z.ZodNumber>;
        stateInputMode: z.ZodOptional<z.ZodEnum<{
            discrete: "discrete";
            continuous: "continuous";
        }>>;
    }, z.core.$strip>;
    backendName: z.ZodNullable<z.ZodString>;
}, z.core.$strip>;
export type VlaHparamsResponse = z.infer<typeof vlaHparamsResponseSchema>;
export interface VlaClientRunParams {
    modelId: string;
    images: Float32Array[];
    imgWidth: number;
    imgHeight: number;
    state: Float32Array;
    tokens: Int32Array;
    mask: Uint8Array;
    noise?: Float32Array;
}
export interface VlaClientRunResult {
    actions: Float32Array;
    actionDim: number;
    chunkSize: number;
    stats?: VlaStats;
}
//# sourceMappingURL=vla.d.ts.map