Efficient-Large-Model/SANA-WM_chunk_causal

SANA-WM Chunk-Causal Stage-1 Teacher

This repository contains the chunk-causal Stage-1 teacher DiT checkpoint for SANA-WM. It is intended to be used with the public SANA-WM code and reuses the VAE and refiner from Efficient-Large-Model/SANA-WM_bidirectional.

Files

• config.yaml: inference config with scheduler.vis_sampler: chunk_flow_euler.
• dit/sana_wm_chunk_causal_1600m_720p.safetensors: chunk-causal Stage-1 teacher checkpoint.

Inference

python inference_video_scripts/wm/inference_sana_wm.py \
  --config hf://Efficient-Large-Model/SANA-WM_chunk_causal/config.yaml \
  --model_path hf://Efficient-Large-Model/SANA-WM_chunk_causal/dit/sana_wm_chunk_causal_1600m_720p.safetensors \
  --image asset/sana_wm/demo_0.png \
  --prompt asset/sana_wm/demo_0.txt \
  --action "w-80,jw-40,w-40,lw-60,w-100" \
  --num_frames 961 \
  --output_dir results/sana_wm_chunk_causal

The sampler uses interval_k = 1 / num_chunks by default.