Skip to content

Commit 182de6a

Browse files
committed
Fix broken VLM snippets on the hub.
1 parent 4e04a5c commit 182de6a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

packages/tasks/src/model-libraries-snippets.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1164,10 +1164,11 @@ export const transformers = (model: ModelData): string[] => {
11641164
].join("\n"),
11651165
"]"
11661166
);
1167+
pipelineSnippet.push("pipe(text=messages)");
11671168
} else {
11681169
pipelineSnippet.push("messages = [", ' {"role": "user", "content": "Who are you?"},', "]");
1170+
pipelineSnippet.push("pipe(messages)");
11691171
}
1170-
pipelineSnippet.push("pipe(messages)");
11711172
}
11721173

11731174
return [pipelineSnippet.join("\n"), autoSnippet];

0 commit comments

Comments
 (0)