Skip to content

Commit dc4226e

Browse files
committed
Trying to save space
tiny is is not so tiny, it's 600M Signed-off-by: Eric Curtin <[email protected]>
1 parent e0e3ee1 commit dc4226e

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

test/system/040-serve.bats

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -178,25 +178,26 @@ verify_begin=".*run --rm"
178178
}
179179

180180
@test "ramalama serve --generate=quadlet" {
181-
model=tiny
181+
model="smollm:135m"
182+
quadlet="$model.container"
182183
name=c_$(safename)
183184
run_ramalama pull ${model}
184185
run_ramalama -q serve --port 1234 --generate=quadlet ${model}
185-
is "$output" "Generating quadlet file: tinyllama.container" "generate tinllama.container"
186+
is "$output" "Generating quadlet file: $quadlet" "generate $quadlet"
186187

187-
run cat tinyllama.container
188+
run cat $quadlet
188189
is "$output" ".*PublishPort=1234:1234" "PublishPort should match"
189190
is "$output" ".*Exec=.*llama-server --port 1234 --model .*" "Exec line should be correct"
190-
is "$output" ".*Mount=type=bind,.*tinyllama" "Mount line should be correct"
191+
is "$output" ".*Mount=type=bind,.*$model" "Mount line should be correct"
191192

192193
HIP_VISIBLE_DEVICES=99 run_ramalama -q serve --port 1234 --generate=quadlet ${model}
193-
is "$output" "Generating quadlet file: tinyllama.container" "generate tinllama.container"
194+
is "$output" "Generating quadlet file: $quadlet.container" "generate $quadlet"
194195

195-
run cat tinyllama.container
196+
run cat $quadlet
196197
is "$output" ".*Environment=HIP_VISIBLE_DEVICES=99" "Should contain env property"
197198

198-
rm tinyllama.container
199-
run_ramalama 2 serve --name=${name} --port 1234 --generate=bogus tiny
199+
rm $quadlet
200+
run_ramalama 2 serve --name=${name} --port 1234 --generate=bogus $model
200201
is "$output" ".*error: argument --generate: invalid choice: .*bogus.* (choose from.*quadlet.*kube.*quadlet/kube.*)" "Should fail"
201202
}
202203

0 commit comments

Comments
 (0)