Modelpublic

NYCU-MLLab/State-Space-Flow-Matching-for-Shortcut-Generation

sourceHugging Faceupdated 1mo agoView on Hugging Face

0likes

Model Card

SSFM Model Variants

This repository contains four SSFM student transformer checkpoints for the Sana 1024px pipeline. The repository contains student transformer weights only; the base pipeline is downloaded separately.

Models

Directory	Variant	Supported steps	Notes
`SSFM_fixed_2`	Fixed-step SSFM	2	EMA student checkpoint
`SSFM_fixed_4`	Fixed-step SSFM	4	EMA student checkpoint
`SSFM_unified_24`	Unified SSFM	2, 4	EMA student checkpoint
`SSFM_unified_124`	Unified SSFM	1, 2, 4	EMA student checkpoint

Base model

Use the official Sana Sprint teacher pipeline:

Efficient-Large-Model/Sana_Sprint_0.6B_1024px_teacher_diffusers

The official base model is not duplicated in this repository.

File layout

Each model directory contains:

text

config.json
diffusion_pytorch_model.safetensors

Use the selected directory as the STUDENT_MODEL_DIR, FIXED2_MODEL_DIR, or FIXED4_MODEL_DIR path in the SSFM GitHub code.

License and attribution

These checkpoints are derived from the Sana/Sana-Sprint model family. Please follow the upstream model license and terms when using or redistributing the weights:

https://huggingface.co/Efficient-Large-Model/SanaSprint0.6B1024pxteacher_diffusers