mirror of
https://github.com/hpcaitech/Open-Sora.git
synced 2026-04-10 21:01:26 +02:00
29 lines
716 B
Bash
29 lines
716 B
Bash
#!/bin/bash
|
|
|
|
set -x
|
|
set -e
|
|
|
|
CKPT=$1
|
|
NUM_FRAMES=$2
|
|
MODEL_NAME=$3
|
|
|
|
if [[ $CKPT == *"ema"* ]]; then
|
|
parentdir=$(dirname $CKPT)
|
|
CKPT_BASE=$(basename $parentdir)_ema
|
|
else
|
|
CKPT_BASE=$(basename $CKPT)
|
|
fi
|
|
LOG_BASE=logs/samples/${MODEL_NAME}_${CKPT_BASE}
|
|
mkdir -p logs/samples
|
|
echo "Logging to $LOG_BASE"
|
|
|
|
GPUS=(0 1 2 3 4 5 6 7)
|
|
TASK_ID_LIST=(1 2a 2b 2c 2d 2e 2f 2g) # 2h not supported yet
|
|
FRAME_LIST=(1 $NUM_FRAMES $NUM_FRAMES $NUM_FRAMES $NUM_FRAMES $NUM_FRAMES $NUM_FRAMES $NUM_FRAMES)
|
|
|
|
for i in "${!GPUS[@]}"; do
|
|
CUDA_VISIBLE_DEVICES=${GPUS[i]} bash eval/sample.sh $CKPT ${FRAME_LIST[i]} $MODEL_NAME -${TASK_ID_LIST[i]} >${LOG_BASE}_${TASK_ID_LIST[i]}.log 2>&1 &
|
|
done
|
|
|
|
# kill all by: pkill -f "inference"
|