meg HF Staff commited on
Commit
e0d94ca
·
verified ·
1 Parent(s): 81672b4

Update entrypoint.sh

Browse files
Files changed (1) hide show
  1. entrypoint.sh +1 -1
entrypoint.sh CHANGED
@@ -4,5 +4,5 @@ echo("Checking h100")
4
  python /check_h100.py
5
  echo("Attempting to run.")
6
  #if [[ $? = 0 ]]; then
7
- optimum-benchmark --config-name image_to_text backend.model=noamrot/FuseCap_Image_Captioning backend.processor=noamrot/FuseCap_Image_Captioning #--config-dir /optimum-benchmark/examples/energy_star/ $@
8
  #fi
 
4
  python /check_h100.py
5
  echo("Attempting to run.")
6
  #if [[ $? = 0 ]]; then
7
+ optimum-benchmark --config-name text_classification #--config-dir /optimum-benchmark/examples/energy_star/ $@
8
  #fi