Cleaning up

Dockerfile

@@ -57,15 +57,12 @@ RUN pip install -r requirements.txt
 
 RUN git clone -b energy_star_dev https://github.com/huggingface/optimum-benchmark.git /optimum-benchmark && cd optimum-benchmark && pip install -e .
 
-COPY ./check_h100.py /check_h100.py
 COPY ./.cache /.cache
 COPY ./entrypoint.sh /entrypoint.sh
 COPY ./pause_space.py /pause_space.py
 COPY ./parse_requests.py /parse_requests.py
 COPY ./create_results.py /create_results.py
-COPY ./failed_run.py /failed_run.py
 COPY ./runs /runs
-COPY ./upload_run_folder.py /upload_run_folder.py
 COPY ./attempts.txt /attempts.txt
 COPY ./failed_attempts.txt /failed_attempts.txt
 

entrypoint.sh

@@ -2,10 +2,7 @@
 
 export SPACE="AIEnergyScore/launch-computation-example"
 
-echo "Not checking h100 -- already know it's not there."
-#python /check_h100.py
 echo "Attempting to run."
-#if [[ $? = 0 ]]; then
 
 # For each line in the requests dataset....
 python /parse_requests.py | while read -r line; do
@@ -21,15 +18,12 @@ python /parse_requests.py | while read -r line; do
 
     # Let the benchmarking begin!
     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" ||
-    echo "${experiment_name},${backend_model}" >> /failed_attempts.txt #(python /failed_run.py --run_dir "${run_dir}" --model_name "${backend_model}" && rm -rf $run_dir)
+    echo "${experiment_name},${backend_model}" >> /failed_attempts.txt
 done
 
 echo "Finished; updating requests dataset and results dataset."
 python /create_results.py ./runs
 
-#echo "Uploading all output from the /runs folder."
-#python /upload_run_folder.py --run_dir "/runs"
-
 # Pausing space
 echo "Pausing space."
 python /pause_space.py