starflow

STARFlow safetensors checkpoints

Safetensors conversions of Apple's STARFlow checkpoints. The files are intended for ComfyUI-STARFlow.

Models

STARFlow 3B — T2I

The conversion script selects all tensor entries from the source state dictionary and skips non-tensor metadata. The source and converted file hashes are recorded above; unlike the T2V conversion, no tensor-by-tensor verification claim is made for this T2I file.

import torch
from safetensors.torch import save_file

src = torch.load("starflow_3B_t2i_256x256.pth", map_location="cpu")
if isinstance(src, dict) and "state_dict" in src:
    src = src["state_dict"]
tensors = {k: v for k, v in src.items()
           if isinstance(k, str) and torch.is_tensor(v)}
save_file(tensors, "starflow_3B_t2i_256x256.safetensors")

Use the STARFlow T2I workflow with stabilityai/sd-vae-ft-ema (or the equivalent ComfyUI SD VAE checkpoint) and a local FLAN-T5-XL text encoder.

STARFlow-V 7B — T2V / first-frame I2V

The conversion uses torch.load(..., weights_only=True) and writes one safetensors file without prefix stripping, dtype conversion, or tensor-value changes. Non-contiguous tensors are made contiguous only as required by the safetensors format.

Use the STARFlow-V T2V or first-frame I2V workflow with the native Wan2.2 VAE.

ComfyUI

The current public ComfyUI-STARFlow release supports the T2I checkpoint. STARFlow-V T2V and first-frame I2V nodes are being integrated into the same repository; use a plugin revision that explicitly includes the STARFlowV* nodes before selecting the video checkpoint. T2V/I2V additionally require a local FLAN-T5-XL text encoder and the Wan2.2-TI2V VAE.

License

Apple Machine Learning Research Model is licensed under the Apple Machine Learning Research Model License Agreement. See LICENSE_MODEL. These files are format-only redistributions/model derivatives: no weights were retrained, fine-tuned, or altered in value. The license permits research purposes only and excludes commercial exploitation, product development, and use in a commercial product or service.