@@ -82,32 +82,28 @@ THREADS_STEP="${THREADS_STEP:-4}"
8282PROMPT=" ${PROMPT:- grab the block.} "
8383case " ${TEST_SUITE} " in
8484 smolvla-libero)
85- DEFAULT_IMAGE_NAMES= " observation.images.image,observation.images.image2"
86- DEFAULT_IMAGE_WIDTH= " 256"
87- DEFAULT_IMAGE_HEIGHT= " 256"
88- DEFAULT_STATE_DIM= " 8 "
85+ IMAGE_NAMES= " ${IMAGE_NAMES :- ${IMAGE_NAME :- observation.images.image,observation.images.image2} } "
86+ IMAGE_WIDTH= " ${IMAGE_WIDTH :- 256} "
87+ IMAGE_HEIGHT= " ${IMAGE_HEIGHT :- 256} "
88+ STATE_DIM= " ${STATE_DIM :- 8} "
8989 ;;
9090 smolvla-so101)
91- DEFAULT_IMAGE_NAMES= " observation.images.front"
92- DEFAULT_IMAGE_WIDTH= " 224"
93- DEFAULT_IMAGE_HEIGHT= " 224"
94- DEFAULT_STATE_DIM= " 6 "
91+ IMAGE_NAMES= " ${IMAGE_NAMES :- ${IMAGE_NAME :- observation.images.front} } "
92+ IMAGE_WIDTH= " ${IMAGE_WIDTH :- 224} "
93+ IMAGE_HEIGHT= " ${IMAGE_HEIGHT :- 224} "
94+ STATE_DIM= " ${STATE_DIM :- 6} "
9595 ;;
9696 pi0-libero)
97- DEFAULT_IMAGE_NAMES= " observation.images.image,observation.images.image2"
98- DEFAULT_IMAGE_WIDTH= " 256"
99- DEFAULT_IMAGE_HEIGHT= " 256"
100- DEFAULT_STATE_DIM= " 32 "
97+ IMAGE_NAMES= " ${IMAGE_NAMES :- ${IMAGE_NAME :- observation.images.image,observation.images.image2} } "
98+ IMAGE_WIDTH= " ${IMAGE_WIDTH :- 256} "
99+ IMAGE_HEIGHT= " ${IMAGE_HEIGHT :- 256} "
100+ STATE_DIM= " ${STATE_DIM :- 32} "
101101 ;;
102102 * )
103103 echo " unsupported TEST_SUITE=${TEST_SUITE} " >&2
104104 exit 1
105105 ;;
106106esac
107- IMAGE_NAMES=" ${IMAGE_NAMES:- ${IMAGE_NAME:- ${DEFAULT_IMAGE_NAMES} } } "
108- IMAGE_WIDTH=" ${IMAGE_WIDTH:- ${DEFAULT_IMAGE_WIDTH} } "
109- IMAGE_HEIGHT=" ${IMAGE_HEIGHT:- ${DEFAULT_IMAGE_HEIGHT} } "
110- STATE_DIM=" ${STATE_DIM:- ${DEFAULT_STATE_DIM} } "
111107WARMUP=" ${WARMUP:- 5} "
112108LOOPS=" ${LOOPS:- 100} "
113109SERVER_WAIT_S=" ${SERVER_WAIT_S:- 120} "
0 commit comments