Apply suggestions from code review

Co-authored-by: vb <vaibhavs10@gmail.com>
huggingface · Oct 7, 2024 · 5246377 · 5246377
1 parent 20fd826
commit 5246377
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/packages/tasks/src/local-apps.ts b/packages/tasks/src/local-apps.ts
@@ -89,7 +89,7 @@ const snippetLlamacpp = (model: ModelData, filepath?: string): LocalAppSnippet[]
 			`${binary} \\`,
 			`  --hf-repo "${model.id}" \\`,
 			`  --hf-file ${filepath ?? "{{GGUF_FILE}}"} \\`,
-			`  -p "${model.tags.includes("conversational") ? "You are a helpful assistant" : "Once upon a time "}"`,
+			`  -p "${model.tags.includes("conversational") ? "You are a helpful assistant" : "Once upon a time"}"`,
 		].join("\n");
 		if (model.tags.includes("conversational")) {
 			snippet += " \\\n  --conversation";
@@ -164,7 +164,7 @@ const snippetVllm = (model: ModelData): LocalAppSnippet[] => {
 		`	-H "Content-Type: application/json" \\`,
 		`	--data '{`,
 		`		"model": "${model.id}",`,
-		`		"prompt": "Once upon a time ",`,
+		`		"prompt": "Once upon a time",`,
 		`		"max_tokens": 512,`,
 		`		"temperature": 0.5`,
 		`	}'`,

diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts
@@ -381,7 +381,7 @@ llm = Llama.from_pretrained(
 )`;
 	} else {
 		snippet += `output = llm(
-	"Once upon a time ",
+	"Once upon a time",
 	max_tokens=512,
 	echo=True
 )