tiny-sd-gguf / run-tiny-sd.sh
turingevo's picture
Upload 3 files
4230e89 verified
#!/bin/bash
# 设置模型路径和输出路径
TYPE='q4_K'
#TYPE='q8_0'
MODEL_DIR="/media/wmx/soft2/huggingface_cache/segmind/tiny-sd-gguf/segmind_tiny-sd-${TYPE}.gguf"
OUTPUT_IMAGE="tiny_sd_output-${TYPE}.png"
# Tiny-SD 本质是极小化的 SD 1.5 架构
# 尺寸必须是 512x512
# 采样器使用默认的 euler_a (或者 dpm++2m)
# CFG 7.0 是 SD 1.5 的标准值
sd-cli \
-m "$MODEL_DIR" \
-W 512 -H 512 \
--cfg-scale 7.0 \
--steps 20 \
--sampling-method euler_a \
--diffusion-fa \
-v \
-o "$OUTPUT_IMAGE" \
-p "a cute car, high quality"
echo ""
if [ ! -f "$OUTPUT_IMAGE" ]; then
echo "[错误] 图像生成失败,请检查上面的错误日志。"
exit 1
fi
echo "生成完毕!图片保存在: $OUTPUT_IMAGE"