| # For licensing see accompanying LICENSE file. | |
| # Copyright (C) 2025 Apple Inc. All Rights Reserved. | |
| export NCCL_DEBUG=WARN | |
| bz=8 | |
| ngpus=1 | |
| cap=${1:-"A man is running"} | |
| input_image=${2:-none} | |
| echo caption=$cap | |
| echo input_image=$input_image | |
| torchrun --standalone --nproc_per_node $ngpus sample.py \ | |
| --model_config_path "configs/starflow_3B_t2i_256x256.yaml" \ | |
| --checkpoint_path "ckpts/starflow_3B_t2i_256x256.pth" \ | |
| --caption "$cap" --sample_batch_size $bz --cfg 3.6 --aspect_ratio "1:1" \ | |
| --seed 999 --save_folder 1 \ | |
| --finetuned_vae none --jacobi 1 --jacobi_th 0.001 --jacobi_block_size 16 |