CONFIG=input/config.json
OUTPUT=/app/output/$(python3 read_json.py $CONFIG task_id)
python3 load_config.py $CONFIG
echo "Convert PTQ YAML Haved been Prepared"




######### Img Adaptor
cd $OUTPUT/Img_Adaptor
BEGIN_IMG_ADAPTOR_TIME=$(date +%s)
echo -e "\033[44;37m===== Start Compiling Img Adaptor =====\033[0m"
hb_compile --config $OUTPUT/img_adaptor.yaml
echo -e "\033[44;37m===== End Compiling Img Adaptor =====\033[0m"
END_IMG_ADAPTOR_TIME=$(date +%s)
IMG_ADAPTOR_TIME=$((END_IMG_ADAPTOR_TIME - BEGIN_IMG_ADAPTOR_TIME))
echo -e "\033[44;37m===== Img Adaptor Time =====\033[0m"
echo -e "\033[44;37m===== $IMG_ADAPTOR_TIME seconds =====\033[0m"

########## DiT
cd $OUTPUT/DiT_Policy
BEGIN_DIT_TIME=$(date +%s)
echo -e "\033[44;37m===== Start Compiling DiT =====\033[0m"
hb_compile --config $OUTPUT/dit.yaml
echo -e "\033[44;37m===== End Compiling DiT =====\033[0m"
END_DIT_TIME=$(date +%s)
DIT_TIME=$((END_DIT_TIME - BEGIN_DIT_TIME))
echo -e "\033[44;37m===== DiT Time =====\033[0m"
echo -e "\033[44;37m===== $DIT_TIME seconds =====\033[0m"