meg's picture
meg HF staff
Debugging getting the error logs.
b634aca verified
raw
history blame
1.53 kB
#!/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 to get the error logs."
curl https://api.hf.space/v1/${SPACE}/logs/run -H "Authorization: Bearer ${DEBUG}"
echo "DID IT WORK?!"
}
#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."