23 lines
642 B
Plaintext
23 lines
642 B
Plaintext
#!/usr/bin/expect
|
|
|
|
set timeout -1
|
|
set host "156.225.27.52"
|
|
set port "5204"
|
|
set user "root"
|
|
set password $env(DEPLOY_SSH_PASSWORD)
|
|
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."
|