bleam 0.0.8 → 0.0.9

package/dist/cli.cjs

@@ -635,8 +635,9 @@ function writeProjectRuntimeConfig(projectRoot, config, origin) {
 	}, null, 2)}\n`);
 	return projectConfigPath;
 }
-function nativeRuntimeTemplateRoot() {
-	return node_path.default.resolve(packageRoot, "..", "..", "app", "bleam");
+function nativeRuntimeTemplateRoot(root = packageRoot) {
+	const bundledTemplate = node_path.default.join(root, "templates", "native");
+	return (0, node_fs.existsSync)(bundledTemplate) ? bundledTemplate : node_path.default.resolve(root, "..", "..", "app", "bleam");
 }
 function safeArtifactName(name) {
 	return name.trim().replace(/[/:\\]/g, "-").replace(/\s+/g, " ") || "App";
@@ -742,8 +743,10 @@ function createBuildNativeWorkspace(options, config) {
 		recursive: true,
 		verbatimSymlinks: true,
 		filter(source) {
-			return !node_path.default.relative(templateRoot, source).split(node_path.default.sep).some((part) => {
-				return part === "build" || part === "dist" || part === "node_modules" || part === ".expo" || part === "Pods" || part === ".xcode.env.local" || part === ".DS_Store" || part.endsWith(".xcworkspace");
+			const parts = node_path.default.relative(templateRoot, source).split(node_path.default.sep);
+			return !parts.some((part, index) => {
+				const frameworkBuildScript = part === "build" && index === 1 && parts[0] === "scripts";
+				return part === "build" && !frameworkBuildScript || part === "dist" || part === "node_modules" || part === ".expo" || part === "Pods" || part === ".xcode.env.local" || part === ".DS_Store" || part.endsWith(".xcworkspace");
 			});
 		}
 	});
@@ -790,6 +793,20 @@ function installPods(nativeRoot) {
 	runInheritedCommand("pod", ["install"], iosRoot);
 	(0, node_fs.writeFileSync)(podInstallStampPath(iosRoot), `${expectedStamp}\n`);
 }
+function ensureNativeFrameworks(nativeRoot) {
+	const frameworksRoot = node_path.default.join(nativeRoot, "ios", "Vendor", "MLXSwift", "0.31.6-macos");
+	if ([
+		"Cmlx",
+		"MLX",
+		"MLXNN",
+		"MLXOptimizers"
+	].every((name) => (0, node_fs.existsSync)(node_path.default.join(frameworksRoot, `${name}.framework`)))) {
+		console.log("Using cached native MLX frameworks");
+		return;
+	}
+	console.log("Preparing native MLX frameworks...");
+	runInheritedCommand(node_path.default.join(resolvedDependencyRoot(), ".bin", "tsx"), [node_path.default.join(nativeRoot, "scripts", "build", "mlx-frameworks.ts")], nativeRoot);
+}
 function podInstallInputStamp(iosRoot) {
 	const inputHash = (0, node_crypto.createHash)("sha256");
 	for (const relative of [
@@ -866,6 +883,7 @@ async function buildNativeArchives(options) {
 	const config = readProjectConfig(options);
 	preflightBuild(options, config);
 	const nativeRoot = createBuildNativeWorkspace(options, config);
+	ensureNativeFrameworks(nativeRoot);
 	installPods(nativeRoot);
 	const archivePath = archiveNativeApp({
 		nativeRoot,
@@ -1883,6 +1901,7 @@ const __test = {
 	hasCurrentPodInstall,
 	installedAppPath,
 	installArchivedApp,
+	nativeRuntimeTemplateRoot,
 	nativeArchiveArgs,
 	parseBuildOptions,
 	parseNewOptions,

package/dist/cli.d.cts

@@ -36,6 +36,7 @@ declare function createProject(options: {
   projectRoot: string;
   template?: ProjectTemplate;
 }): void;
+declare function nativeRuntimeTemplateRoot(root?: string): string;
 declare function replaceOrInsertPlistString(plist: string, key: string, value: string): string;
 declare function createBuildNativeWorkspace(options: BuildOptions, config: ProjectConfig): string;
 declare function podInstallInputStamp(iosRoot: string): string;
@@ -77,6 +78,7 @@ declare const __test: {
   hasCurrentPodInstall: typeof hasCurrentPodInstall;
   installedAppPath: typeof installedAppPath;
   installArchivedApp: typeof installArchivedApp;
+  nativeRuntimeTemplateRoot: typeof nativeRuntimeTemplateRoot;
   nativeArchiveArgs: typeof nativeArchiveArgs;
   parseBuildOptions: typeof parseBuildOptions;
   parseNewOptions: typeof parseNewOptions;

package/dist/cli.d.ts

@@ -36,6 +36,7 @@ declare function createProject(options: {
   projectRoot: string;
   template?: ProjectTemplate;
 }): void;
+declare function nativeRuntimeTemplateRoot(root?: string): string;
 declare function replaceOrInsertPlistString(plist: string, key: string, value: string): string;
 declare function createBuildNativeWorkspace(options: BuildOptions, config: ProjectConfig): string;
 declare function podInstallInputStamp(iosRoot: string): string;
@@ -77,6 +78,7 @@ declare const __test: {
   hasCurrentPodInstall: typeof hasCurrentPodInstall;
   installedAppPath: typeof installedAppPath;
   installArchivedApp: typeof installArchivedApp;
+  nativeRuntimeTemplateRoot: typeof nativeRuntimeTemplateRoot;
   nativeArchiveArgs: typeof nativeArchiveArgs;
   parseBuildOptions: typeof parseBuildOptions;
   parseNewOptions: typeof parseNewOptions;

package/dist/cli.js

@@ -623,8 +623,9 @@ function writeProjectRuntimeConfig(projectRoot, config, origin) {
 	}, null, 2)}\n`);
 	return projectConfigPath;
 }
-function nativeRuntimeTemplateRoot() {
-	return path.resolve(packageRoot, "..", "..", "app", "bleam");
+function nativeRuntimeTemplateRoot(root = packageRoot) {
+	const bundledTemplate = path.join(root, "templates", "native");
+	return existsSync(bundledTemplate) ? bundledTemplate : path.resolve(root, "..", "..", "app", "bleam");
 }
 function safeArtifactName(name) {
 	return name.trim().replace(/[/:\\]/g, "-").replace(/\s+/g, " ") || "App";
@@ -730,8 +731,10 @@ function createBuildNativeWorkspace(options, config) {
 		recursive: true,
 		verbatimSymlinks: true,
 		filter(source) {
-			return !path.relative(templateRoot, source).split(path.sep).some((part) => {
-				return part === "build" || part === "dist" || part === "node_modules" || part === ".expo" || part === "Pods" || part === ".xcode.env.local" || part === ".DS_Store" || part.endsWith(".xcworkspace");
+			const parts = path.relative(templateRoot, source).split(path.sep);
+			return !parts.some((part, index) => {
+				const frameworkBuildScript = part === "build" && index === 1 && parts[0] === "scripts";
+				return part === "build" && !frameworkBuildScript || part === "dist" || part === "node_modules" || part === ".expo" || part === "Pods" || part === ".xcode.env.local" || part === ".DS_Store" || part.endsWith(".xcworkspace");
 			});
 		}
 	});
@@ -778,6 +781,20 @@ function installPods(nativeRoot) {
 	runInheritedCommand("pod", ["install"], iosRoot);
 	writeFileSync(podInstallStampPath(iosRoot), `${expectedStamp}\n`);
 }
+function ensureNativeFrameworks(nativeRoot) {
+	const frameworksRoot = path.join(nativeRoot, "ios", "Vendor", "MLXSwift", "0.31.6-macos");
+	if ([
+		"Cmlx",
+		"MLX",
+		"MLXNN",
+		"MLXOptimizers"
+	].every((name) => existsSync(path.join(frameworksRoot, `${name}.framework`)))) {
+		console.log("Using cached native MLX frameworks");
+		return;
+	}
+	console.log("Preparing native MLX frameworks...");
+	runInheritedCommand(path.join(resolvedDependencyRoot(), ".bin", "tsx"), [path.join(nativeRoot, "scripts", "build", "mlx-frameworks.ts")], nativeRoot);
+}
 function podInstallInputStamp(iosRoot) {
 	const inputHash = createHash("sha256");
 	for (const relative of [
@@ -854,6 +871,7 @@ async function buildNativeArchives(options) {
 	const config = readProjectConfig(options);
 	preflightBuild(options, config);
 	const nativeRoot = createBuildNativeWorkspace(options, config);
+	ensureNativeFrameworks(nativeRoot);
 	installPods(nativeRoot);
 	const archivePath = archiveNativeApp({
 		nativeRoot,
@@ -1871,6 +1889,7 @@ const __test = {
 	hasCurrentPodInstall,
 	installedAppPath,
 	installArchivedApp,
+	nativeRuntimeTemplateRoot,
 	nativeArchiveArgs,
 	parseBuildOptions,
 	parseNewOptions,

package/dist/{ui-Ce-pnUUA.d.ts → ui-Bg11tvlc.d.ts}

@@ -1,4 +1,4 @@
-import * as _nativescript_react_native33 from "@nativescript/react-native";
+import * as _nativescript_react_native27 from "@nativescript/react-native";
 import { ReactNode } from "react";
 import { ColorValue, StyleProp, TextStyle, ViewStyle } from "react-native";
 
@@ -18,7 +18,7 @@ type ButtonProps = {
   onPress?: () => void;
   style?: StyleProp<NativeStyle>;
 };
-declare const Button: _nativescript_react_native33.UIKitViewComponent<ButtonProps, UIButton>;
+declare const Button: _nativescript_react_native27.UIKitViewComponent<ButtonProps, UIButton>;
 //#endregion
 //#region src/ui/label.d.ts
 type LabelTone = 'primary' | 'secondary' | 'tertiary';
@@ -29,7 +29,7 @@ type LabelProps = {
   numberOfLines?: number;
   style?: StyleProp<NativeStyle>;
 };
-declare const Label: _nativescript_react_native33.UIKitViewComponent<LabelProps, UILabel>;
+declare const Label: _nativescript_react_native27.UIKitViewComponent<LabelProps, UILabel>;
 //#endregion
 //#region src/ui/symbol.d.ts
 type SymbolName = string;
@@ -43,7 +43,7 @@ type SymbolProps = {
   color?: ColorValue;
   style?: StyleProp<NativeStyle>;
 };
-declare const Symbol: _nativescript_react_native33.UIKitViewComponent<SymbolProps, UIImageView>;
+declare const Symbol: _nativescript_react_native27.UIKitViewComponent<SymbolProps, UIImageView>;
 //#endregion
 //#region src/ui/glass-shared.d.ts
 type GlassStyle = 'regular' | 'clear' | 'none';
@@ -73,10 +73,10 @@ type GlassContainerProps = {
 };
 //#endregion
 //#region src/ui/glass-view.d.ts
-declare const GlassView: _nativescript_react_native33.UIKitViewComponent<GlassViewProps, _nativescript_react_native33.UIKitContainerResult<UIVisualEffectView, UIView>>;
+declare const GlassView: _nativescript_react_native27.UIKitViewComponent<GlassViewProps, _nativescript_react_native27.UIKitContainerResult<UIVisualEffectView, UIView>>;
 //#endregion
 //#region src/ui/glass-container.d.ts
-declare const GlassContainer: _nativescript_react_native33.UIKitViewComponent<GlassContainerProps, _nativescript_react_native33.UIKitContainerResult<UIVisualEffectView, UIView>>;
+declare const GlassContainer: _nativescript_react_native27.UIKitViewComponent<GlassContainerProps, _nativescript_react_native27.UIKitContainerResult<UIVisualEffectView, UIView>>;
 //#endregion
 //#region src/ui/blur-view.d.ts
 type BlurIntensity = 'ultraThin' | 'thin' | 'regular' | 'thick' | 'chrome';
@@ -87,6 +87,6 @@ type BlurViewProps = {
   intensity?: BlurIntensity;
   colorScheme?: BlurColorScheme;
 };
-declare const BlurView: _nativescript_react_native33.UIKitViewComponent<BlurViewProps, _nativescript_react_native33.UIKitContainerResult<UIVisualEffectView, UIView>>;
+declare const BlurView: _nativescript_react_native27.UIKitViewComponent<BlurViewProps, _nativescript_react_native27.UIKitContainerResult<UIVisualEffectView, UIView>>;
 //#endregion
 export { ButtonVariant as C, ButtonSize as S, Label as _, GlassContainer as a, Button as b, GlassEffectStyleConfig as c, GlassVisualStyle as d, Symbol as f, SymbolWeight as g, SymbolScale as h, BlurViewProps as i, GlassStyle as l, SymbolProps as m, BlurIntensity as n, GlassView as o, SymbolName as p, BlurView as r, GlassContainerProps as s, BlurColorScheme as t, GlassViewProps as u, LabelProps as v, ButtonProps as x, LabelTone as y };

package/dist/ui.d.ts

@@ -1,2 +1,2 @@
-import { C as ButtonVariant, S as ButtonSize, _ as Label, a as GlassContainer, b as Button, c as GlassEffectStyleConfig, d as GlassVisualStyle, f as Symbol, g as SymbolWeight, h as SymbolScale, i as BlurViewProps, l as GlassStyle, m as SymbolProps, n as BlurIntensity, o as GlassView, p as SymbolName, r as BlurView, s as GlassContainerProps, t as BlurColorScheme, u as GlassViewProps, v as LabelProps, x as ButtonProps, y as LabelTone } from "./ui-Ce-pnUUA.js";
+import { C as ButtonVariant, S as ButtonSize, _ as Label, a as GlassContainer, b as Button, c as GlassEffectStyleConfig, d as GlassVisualStyle, f as Symbol, g as SymbolWeight, h as SymbolScale, i as BlurViewProps, l as GlassStyle, m as SymbolProps, n as BlurIntensity, o as GlassView, p as SymbolName, r as BlurView, s as GlassContainerProps, t as BlurColorScheme, u as GlassViewProps, v as LabelProps, x as ButtonProps, y as LabelTone } from "./ui-Bg11tvlc.js";
 export { BlurColorScheme, BlurIntensity, BlurView, BlurViewProps, Button, ButtonProps, ButtonSize, ButtonVariant, GlassContainer, GlassContainerProps, GlassEffectStyleConfig, GlassStyle, GlassView, GlassViewProps, GlassVisualStyle, Label, LabelProps, LabelTone, Symbol, SymbolName, SymbolProps, SymbolScale, SymbolWeight };

package/dist/window.d.ts

@@ -1,4 +1,4 @@
-import { p as SymbolName } from "./ui-Ce-pnUUA.js";
+import { p as SymbolName } from "./ui-Bg11tvlc.js";
 import * as react3 from "react";
 import { ComponentType } from "react";
 import { StyleProp, ViewStyle } from "react-native";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "bleam",
-  "version": "0.0.8",
+  "version": "0.0.9",
   "packageManager": "yarn@1.22.22",
   "type": "module",
   "files": [
@@ -15,6 +15,9 @@
   },
   "scripts": {
     "build": "tsdown",
+    "prepublishOnly": "tsdown",
+    "prepack": "node scripts/prepare-native-template.mjs",
+    "postpack": "node scripts/clean-native-template.mjs",
     "test": "vitest run",
     "typecheck": "tsc --noEmit -p tsconfig.build.json"
   },

package/templates/image-generation/app/index.tsx

@@ -16,7 +16,7 @@ export default function App() {
     try {
       const result = await generateImage({
         prompt: 'A precise editorial illustration of a small native Mac app',
-        aspectRatio: 'square',
+        aspectRatio: '1:1',
       })
       setUri(result.uri)
       setStatus(`Completed with seed ${result.metadata.seed}`)

package/templates/native/.gitattributes

@@ -0,0 +1 @@
+*.pbxproj -text

package/templates/native/App.tsx

@@ -0,0 +1,26 @@
+import NativeScript from '@nativescript/react-native'
+import { StyleSheet, Text, View } from 'react-native'
+
+NativeScript.init()
+
+export default function App() {
+  return (
+    <View style={styles.canvas}>
+      <Text style={styles.title}>Bleam</Text>
+      <Text style={styles.message}>
+        Run a Bleam project with npx bleam new or bleam dev.
+      </Text>
+    </View>
+  )
+}
+
+const styles = StyleSheet.create({
+  canvas: {
+    flex: 1,
+    alignItems: 'center',
+    justifyContent: 'center',
+    backgroundColor: '#0b0b0b',
+  },
+  title: { color: '#f5f5f5', fontSize: 28, fontWeight: '700' },
+  message: { color: '#888', fontSize: 14, marginTop: 10 },
+})

package/templates/native/app.json

@@ -0,0 +1,12 @@
+{
+  "expo": {
+    "name": "Bleam",
+    "slug": "bleam",
+    "version": "1.0.0",
+    "platforms": ["ios"],
+    "ios": {
+      "appleTeamId": "5SZ3H5P69M",
+      "bundleIdentifier": "dev.bleam.app"
+    }
+  }
+}

package/templates/native/index.ts

@@ -0,0 +1,8 @@
+import { registerRootComponent } from 'expo'
+
+import App from './App'
+
+// registerRootComponent calls AppRegistry.registerComponent('main', () => App);
+// It also ensures that whether you load the app in Expo Go or in a native build,
+// the environment is set up appropriately
+registerRootComponent(App)

package/templates/native/ios/.xcode.env

@@ -0,0 +1,11 @@
+# This `.xcode.env` file is versioned and is used to source the environment
+# used when running script phases inside Xcode.
+# To customize your local environment, you can create an `.xcode.env.local`
+# file that is not versioned.
+
+# NODE_BINARY variable contains the PATH to the node executable.
+#
+# Customize the NODE_BINARY variable here.
+# For example, to use nvm with brew, add the following line
+# . "$(brew --prefix nvm)/nvm.sh" --no-use
+export NODE_BINARY=$(command -v node)

package/templates/native/ios/Bleam/AI/Flux2/Configuration/Flux2Config.swift

@@ -0,0 +1,288 @@
+// Flux2Config.swift - Flux.2 Transformer Configuration
+// Copyright 2025 Vincent Gourbin
+
+import Foundation
+
+// MARK: - Model Selection
+
+/// Flux.2 model variants
+public enum Flux2Model: String, CaseIterable, Sendable {
+    /// Flux.2 Klein 4B - 4B parameters, Qwen3-4B text encoder (Apache 2.0)
+    case klein4B = "klein-4b"
+    /// Flux.2 Klein 9B - 9B parameters, Qwen3-8B text encoder (Apache 2.0)
+    case klein9B = "klein-9b"
+
+    public var displayName: String {
+        switch self {
+        case .klein4B: return "Flux.2 Klein 4B"
+        case .klein9B: return "Flux.2 Klein 9B"
+        }
+    }
+
+    /// Whether this is a base (non-distilled) model for training
+    public var isBaseModel: Bool {
+        false
+    }
+
+    /// Whether this model can be used for inference
+    public var isForInference: Bool {
+        true
+    }
+
+    /// Whether this model can be used for LoRA training
+    public var isForTraining: Bool {
+        false
+    }
+
+    /// Get the base (non-distilled) variant for training, if available
+    public var trainingVariant: Flux2Model {
+        .klein4B
+    }
+
+    /// Get the distilled variant for inference (validation images during training)
+    public var inferenceVariant: Flux2Model {
+        .klein4B
+    }
+
+    /// Whether this model uses guidance embeddings
+    public var usesGuidanceEmbeds: Bool {
+        false
+    }
+
+    /// Joint attention dimension (text encoder output)
+    public var jointAttentionDim: Int {
+        switch self {
+        case .klein4B: return 7680   // Qwen3-4B: 3 × 2560
+        case .klein9B: return 12288  // Qwen3-8B: 3 × 4096
+        }
+    }
+
+    /// Get the transformer configuration for this model
+    public var transformerConfig: Flux2TransformerConfig {
+        switch self {
+        case .klein4B: return .klein4B
+        case .klein9B: return .klein9B
+        }
+    }
+
+    /// Estimated VRAM usage in GB
+    public var estimatedVRAM: Int {
+        switch self {
+        case .klein4B: return 13
+        case .klein9B: return 20
+        }
+    }
+
+    /// License information
+    public var license: String {
+        "Apache 2.0"
+    }
+
+    /// Whether this model can be used commercially
+    public var isCommercialUseAllowed: Bool {
+        true
+    }
+
+    /// Recommended number of inference steps for optimal quality
+    public var defaultSteps: Int {
+        4
+    }
+
+    /// Recommended guidance scale for optimal quality
+    public var defaultGuidance: Float {
+        1.0
+    }
+
+    /// Estimated generation time in seconds (1024x1024 on M2 Max)
+    public var estimatedTimeSeconds: Int {
+        switch self {
+        case .klein4B: return 26
+        case .klein9B: return 45
+        }
+    }
+
+    /// Maximum number of reference images for I2I generation
+    /// Source: https://docs.bfl.ai/flux_2/flux2_image_editing
+    public var maxReferenceImages: Int {
+        4
+    }
+}
+
+// MARK: - Transformer Configuration
+
+/// Configuration for the Flux.2 diffusion transformer
+public struct Flux2TransformerConfig: Codable, Sendable {
+    /// Patch size for input embedding (1 for Flux.2)
+    public var patchSize: Int
+
+    /// Number of input channels (128 for Flux.2 latents)
+    public var inChannels: Int
+
+    /// Number of output channels (same as input)
+    public var outChannels: Int
+
+    /// Number of double-stream transformer blocks
+    public var numLayers: Int
+
+    /// Number of single-stream transformer blocks
+    public var numSingleLayers: Int
+
+    /// Dimension of each attention head
+    public var attentionHeadDim: Int
+
+    /// Number of attention heads
+    public var numAttentionHeads: Int
+
+    /// Inner dimension for transformer (numAttentionHeads * attentionHeadDim)
+    public var innerDim: Int {
+        numAttentionHeads * attentionHeadDim
+    }
+
+    /// Dimension of joint attention (from Mistral embeddings: 15360)
+    public var jointAttentionDim: Int
+
+    /// Dimension of pooled projection (time + guidance embeddings)
+    public var pooledProjectionDim: Int
+
+    /// Whether to use guidance embedding
+    public var guidanceEmbeds: Bool
+
+    /// Axes dimensions for RoPE [T, H, W, L]
+    public var axesDimsRope: [Int]
+
+    /// Base theta for RoPE
+    public var ropeTheta: Float
+
+    /// MLP expansion ratio (3.0 for Flux.2, determines FFN hidden dimension)
+    public var mlpRatio: Float
+
+    /// Activation function for feedforward
+    public var activationFunction: String
+
+    public init(
+        patchSize: Int = 1,
+        inChannels: Int = 128,
+        outChannels: Int = 128,
+        numLayers: Int = 8,
+        numSingleLayers: Int = 48,
+        attentionHeadDim: Int = 128,
+        numAttentionHeads: Int = 48,
+        jointAttentionDim: Int = 15360,
+        pooledProjectionDim: Int = 768,
+        guidanceEmbeds: Bool = true,
+        axesDimsRope: [Int] = [32, 32, 32, 32],
+        ropeTheta: Float = 2000.0,
+        mlpRatio: Float = 3.0,
+        activationFunction: String = "silu"
+    ) {
+        self.patchSize = patchSize
+        self.inChannels = inChannels
+        self.outChannels = outChannels
+        self.numLayers = numLayers
+        self.numSingleLayers = numSingleLayers
+        self.attentionHeadDim = attentionHeadDim
+        self.numAttentionHeads = numAttentionHeads
+        self.jointAttentionDim = jointAttentionDim
+        self.pooledProjectionDim = pooledProjectionDim
+        self.guidanceEmbeds = guidanceEmbeds
+        self.axesDimsRope = axesDimsRope
+        self.ropeTheta = ropeTheta
+        self.mlpRatio = mlpRatio
+        self.activationFunction = activationFunction
+    }
+
+    /// Flux.2 Klein 4B configuration
+    /// 4B parameters, 5 double + 20 single blocks
+    public static let klein4B = Flux2TransformerConfig(
+        patchSize: 1,
+        inChannels: 128,
+        outChannels: 128,
+        numLayers: 5,
+        numSingleLayers: 20,
+        attentionHeadDim: 128,
+        numAttentionHeads: 24,      // 24 × 128 = 3072
+        jointAttentionDim: 7680,    // Qwen3-4B: 3 × 2560
+        pooledProjectionDim: 768,
+        guidanceEmbeds: false,
+        axesDimsRope: [32, 32, 32, 32],
+        ropeTheta: 2000.0,
+        mlpRatio: 3.0,
+        activationFunction: "silu"
+    )
+
+    /// Flux.2 Klein 9B configuration
+    /// 9B parameters, 8 double + 24 single blocks
+    public static let klein9B = Flux2TransformerConfig(
+        patchSize: 1,
+        inChannels: 128,
+        outChannels: 128,
+        numLayers: 8,
+        numSingleLayers: 24,
+        attentionHeadDim: 128,
+        numAttentionHeads: 32,      // 32 × 128 = 4096
+        jointAttentionDim: 12288,   // Qwen3-8B: 3 × 4096
+        pooledProjectionDim: 768,
+        guidanceEmbeds: false,
+        axesDimsRope: [32, 32, 32, 32],
+        ropeTheta: 2000.0,
+        mlpRatio: 3.0,
+        activationFunction: "silu"
+    )
+
+    // MARK: - Codable
+
+    enum CodingKeys: String, CodingKey {
+        case patchSize = "patch_size"
+        case inChannels = "in_channels"
+        case outChannels = "out_channels"
+        case numLayers = "num_layers"
+        case numSingleLayers = "num_single_layers"
+        case attentionHeadDim = "attention_head_dim"
+        case numAttentionHeads = "num_attention_heads"
+        case jointAttentionDim = "joint_attention_dim"
+        case pooledProjectionDim = "pooled_projection_dim"
+        case guidanceEmbeds = "guidance_embeds"
+        case axesDimsRope = "axes_dims_rope"
+        case ropeTheta = "rope_theta"
+        case mlpRatio = "mlp_ratio"
+        case activationFunction = "activation_function"
+    }
+
+    public init(from decoder: Decoder) throws {
+        let container = try decoder.container(keyedBy: CodingKeys.self)
+
+        patchSize = try container.decodeIfPresent(Int.self, forKey: .patchSize) ?? 1
+        inChannels = try container.decodeIfPresent(Int.self, forKey: .inChannels) ?? 128
+        outChannels = try container.decodeIfPresent(Int.self, forKey: .outChannels) ?? 128
+        numLayers = try container.decodeIfPresent(Int.self, forKey: .numLayers) ?? 8
+        numSingleLayers = try container.decodeIfPresent(Int.self, forKey: .numSingleLayers) ?? 48
+        attentionHeadDim = try container.decodeIfPresent(Int.self, forKey: .attentionHeadDim) ?? 128
+        numAttentionHeads = try container.decodeIfPresent(Int.self, forKey: .numAttentionHeads) ?? 48
+        jointAttentionDim = try container.decodeIfPresent(Int.self, forKey: .jointAttentionDim) ?? 15360
+        pooledProjectionDim = try container.decodeIfPresent(Int.self, forKey: .pooledProjectionDim) ?? 768
+        guidanceEmbeds = try container.decodeIfPresent(Bool.self, forKey: .guidanceEmbeds) ?? true
+        axesDimsRope = try container.decodeIfPresent([Int].self, forKey: .axesDimsRope) ?? [32, 32, 32, 32]
+        ropeTheta = try container.decodeIfPresent(Float.self, forKey: .ropeTheta) ?? 2000.0
+        mlpRatio = try container.decodeIfPresent(Float.self, forKey: .mlpRatio) ?? 3.0
+        activationFunction = try container.decodeIfPresent(String.self, forKey: .activationFunction) ?? "silu"
+    }
+
+    /// Load configuration from a JSON file
+    public static func load(from url: URL) throws -> Flux2TransformerConfig {
+        let data = try Data(contentsOf: url)
+        return try JSONDecoder().decode(Flux2TransformerConfig.self, from: data)
+    }
+}
+
+extension Flux2TransformerConfig: CustomStringConvertible {
+    public var description: String {
+        """
+        Flux2TransformerConfig(
+            layers: \(numLayers) double + \(numSingleLayers) single,
+            heads: \(numAttentionHeads) × \(attentionHeadDim) = \(innerDim),
+            jointDim: \(jointAttentionDim),
+            rope: \(axesDimsRope) θ=\(ropeTheta)
+        )
+        """
+    }
+}