2
0
Эх сурвалжийг харах

fill in the middle example

Bruce MacDonald 1 жил өмнө
parent
commit
99a8cf3b7d

+ 16 - 0
examples/fill-in-middle/fill.ts

@@ -0,0 +1,16 @@
+import { Ollama } from '../../src/index';
+
+async function main(): Promise<void> {
+    const prefix = `def remove_non_ascii(s: str) -> str:
+"""
+`;
+    const suffix = `
+return result
+`;
+
+    const respose = await new Ollama().generate({model: "codellama:7b-code", prompt: `<PRE> ${prefix} <SUF>${suffix} <MID>`, options: {num_predict: 128, temperature: 0, top_p: 0.9, presence_penalty: 0, stop: ["<EOT>"]}});
+    console.log(respose.response);
+
+}
+
+await main();