exec_path="$1"

if [ -n "$2" ]; then
  rank_to_worldsize=$2-$(nvidia-smi -L | wc -l)
  echo running on rank $2, input will be partitioned by $rank_to_worldsize
  export CUDA_VISIBLE_DEVICES=$2
else
  rank_to_worldsize='-'
fi

echo $rank_to_worldsize

# "parse_live2d", "--exec_list", "workspace/datasets/model_pack2_jxl_crop_annotated/exec_list.txt", "--detected_instanec_only"

echo running infer_langsam for $exec_path
conda run --no-capture-output -n see_through_dev python inference/scripts/parse_live2d.py infer_langsam --exec_list $exec_path --rank_to_worldsize $rank_to_worldsize --detected_instanec_only --skip_exists