Update examples/llava/mtmd-cli.cpp

ngxson · CISC · web-flow · commit 9384cb4b5ee5 · 2025-05-01T23:16:06.000+02:00
Co-authored-by: Sigbjørn Skjæret &lt;sigbjorn.skjaeret@scala.com&gt;
diff --git a/examples/llava/mtmd-cli.cpp b/examples/llava/mtmd-cli.cpp
@@ -316,7 +316,7 @@ int main(int argc, char ** argv) {
                 continue;
             }
             g_is_generating = true;
-            if (line.find("/image") == 0) {
+            if (line == "/image" || line.find("/image ") == 0) {
                 if (line.size() < 8) {
                     LOG_ERR("ERR: Missing image filename\n");
                     continue;

Original file line number	Diff line number	Diff line change
`@@ -316,7 +316,7 @@ int main(int argc, char ** argv) {`
`316`	`316`	`continue;`
`317`	`317`	`}`
`318`	`318`	`g_is_generating = true;`
`319`		`- if (line.find("/image") == 0) {`
	`319`	`+ if (line == "/image" \|\| line.find("/image ") == 0) {`
`320`	`320`	`if (line.size() < 8) {`
`321`	`321`	`LOG_ERR("ERR: Missing image filename\n");`
`322`	`322`	`continue;`