File size: 642 Bytes
2cf4a54 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | #!/bin/bash
# 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 |