#!/usr/bin/expect

set timeout -1
set host "156.225.27.52"
set port "5204"
set user "root"
set password "Jsjj@8888"
set local_model_dir "FLUX_Model"
set remote_base_dir "/root/ai-projects/DreamOmni2/checkpoints"

puts "--> Uploading FLUX Model Weights (supporting resume)..."
# -a: archive, -v: verbose, -z: compress, -P: partial/progress
# 上传到 checkpoints 目录下一级
spawn rsync -avzP -e "ssh -p $port" $local_model_dir $user@$host:$remote_base_dir/
expect {
    "password:" { send "$password\r" }
    "yes/no" { send "yes\r"; exp_continue }
}
# Expect EOF or timeout
expect eof

puts "FLUX model upload completed."