Spaces:
Paused
Paused
Update entrypoint.sh
Browse files- entrypoint.sh +4 -4
entrypoint.sh
CHANGED
@@ -4,10 +4,10 @@ config_dir="/optimum-benchmark/examples/energy_star/"
|
|
4 |
|
5 |
# This script is meant to be called from a python script \
|
6 |
# that provides the REPO_ID as the first argument.
|
7 |
-
REPO_ID
|
8 |
-
MODEL
|
9 |
-
TASK
|
10 |
-
HARDWARE
|
11 |
echo "Attempting to run."
|
12 |
# Read the name of the model and the experiment.
|
13 |
echo "Benchmarking Model: ${MODEL}, Task: ${TASK}, Hardware: ${HARDWARE}"
|
|
|
4 |
|
5 |
# This script is meant to be called from a python script \
|
6 |
# that provides the REPO_ID as the first argument.
|
7 |
+
REPO_ID=$1
|
8 |
+
MODEL=$2
|
9 |
+
TASK=$3
|
10 |
+
HARDWARE=$4
|
11 |
echo "Attempting to run."
|
12 |
# Read the name of the model and the experiment.
|
13 |
echo "Benchmarking Model: ${MODEL}, Task: ${TASK}, Hardware: ${HARDWARE}"
|