rlx-flow 0.2.1

Block assembly-line API for RLX model builders — fusion-first, config-driven
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

// RLX — versatile ML compiler + runtime.
// Copyright (C) 2026 Eugene Hauptmann, Nataliya Kosmyna.

//! Composable model recipes — arch presets that remain patchable.

use crate::flow::ModelFlow;

/// Assemble a [`ModelFlow`] from config — use for arch-specific presets (LLaMA, Qwen, FLUX, …).
///
/// Recipes return an unbuilt flow so callers can still `.raw_stage()`, `.custom()`, or
/// `.patch()` before `build()`.
pub trait ModelRecipe {
    fn name(&self) -> &str;
    fn assemble(&self) -> ModelFlow;
}

impl<F> ModelRecipe for F
where
    F: Fn() -> ModelFlow,
{
    fn name(&self) -> &str {
        "closure_recipe"
    }

    fn assemble(&self) -> ModelFlow {
        self()
    }
}