entrypoint.sh

#!/bin/bash

# TODO: I think this flag is related to getting fail logs; check.
set -e

export SPACE="EnergyStarAI/launch-computation-example"

echo "Not checking h100 -- already know it's not there."
#python /check_h100.py
echo "Attempting to run."
#if [[ $? = 0 ]]; then

python /parse_requests.py | while read line; do
    IFS="," read backend_model experiment_name <<< $(echo ${line})
    echo "Benchmarking Model: ${backend_model}, Task: ${experiment_name}"
    export run_dir="./runs/" #${experiment_name}/${backend_model}/$(date +%Y-%m-%d-%H-%M-%S)"

    optimum-benchmark --config-name ${experiment_name}  --config-dir /optimum-benchmark/examples/energy_star/ backend.model=${backend_model} backend.processor=${backend_model} hydra.run.dir=${run_dir} 2> $run_dir/error.log

done || {
    echo "Trying the curl thing."
    curl https://api.hf.space/v1/${SPACE}/logs/run 
    }

    #if [ -s $run_dir/error.log ]; then
    #    # error.log is not-empty, an error was raised
    #    echo "An error was raised while benchmarking the model..."
    #    python /failed_run.py --run_dir $run_dir --model_name $backend_model
    #
    #    # Delete the current run directory so that it is not pushed by create_results.py later
    #    rm -rf $run_dir
    #fi

# The file is empty, so no error
echo "Finished; uploading dataset results"
python /create_results.py ./runs

# Pausing space
echo "Pausing space."
python /pause_space.py
echo "Done."