FLUX.2-klein-4B-CoreAI

FLUX.2 klein 4B — Core AI

Apple Core AI 格式的 FLUX.2 klein 4B， 可在 macOS 26 / iOS 26 及以上直接经 CoreAIDiffusion 加载运行。

⚠️ 本仓库是修改后的衍生作品（Apache License 2.0 §4(b) 声明）

原始权重由 Black Forest Labs 以 Apache License 2.0 发布。本仓库内的文件已被修改， 具体变更如下：

未对模型架构或训练权重的数值语义做任何人为调整，全部变更均由 Apple 官方导出工具产生。

来源与可复现性

任何人都可用上述命令复现出等价产物。

内容

Transformer 的 8 个入口：main、half、img2img_{quarter,half,full}、 img2img_512_{quarter,half,full} —— 共享同一份权重。

用法

import CoreAIDiffusionPipeline

let pipeline = try await Flux2Pipeline(from: modelURL)
let config = PipelineConfiguration(
    prompt: "a calico cat wearing a tiny wizard hat",
    seed: 42, stepCount: 4, guidanceScale: 1.0,
    schedulerType: .discreteFlow)
let result = try await pipeline.generateImages(configuration: config)

图生图额外设置 startingImage 与 referenceGrid（.full / .half / .quarter）。 注意 FLUX.2 忽略 strength，引导强度由 referenceGrid 决定。

实测性能（Apple M1 Max / 64 GB / macOS 27.0）

首次运行约 32.5 s —— Core AI 需在设备端编译一次， 并在 ~/Library/Caches/coreai-cache/ 生成约 3.8 GB 特化缓存；之后每次约 21 s。 系统大版本升级后缓存会失效并重新编译。

已知限制

• 仅支持 1024×1024 与 512×512 两档（导出时固定）
• 图生图与 tiled decode 互斥（VAE 编码器形状不匹配，会抛错）
• 不支持 LoRA —— .aimodel 的权重已编入编译产物

许可

Apache License 2.0，继承自原始模型。完整文本见本仓库 LICENSE。

原始模型版权归 Black Forest Labs 所有。