#!/bin/bash
set -e

echo "================================================"
echo "   FLUX 模型下载与部署脚本"
echo "================================================"

# 1. Download Model
echo "[1/2] 下载 FLUX 模型权重 (Hugging Face)..."
if [ -f "./local_deploy_env/bin/python3" ]; then
    ./local_deploy_env/bin/python3 download_flux.py
else
    python3 download_flux.py
fi

# 2. Upload
echo "[2/2] 上传 FLUX 模型到服务器..."
chmod +x upload_flux.exp
./upload_flux.exp

echo "================================================"
echo "   FLUX 模型部署结束！"
echo "================================================"