fix (#5)

- fix (7aab05e6d84414909b031e6cb7884c2231a6a89c)


Co-authored-by: Radamés Ajna <[email protected]>

build/m.d.ts

@@ -16,11 +16,12 @@ export class Model {
 /**
 * @param {string} prompt
 * @param {number} temp
+* @param {number} top_p
 * @param {number} repeat_penalty
 * @param {bigint} seed
 * @returns {string}
 */
-  init_with_prompt(prompt: string, temp: number, repeat_penalty: number, seed: bigint): string;
+  init_with_prompt(prompt: string, temp: number, top_p: number, repeat_penalty: number, seed: bigint): string;
 /**
 * @returns {string}
 */
@@ -34,7 +35,7 @@ export interface InitOutput {
   readonly __wbg_model_free: (a: number) => void;
   readonly model_new: (a: number, b: number, c: number, d: number, e: number) => void;
   readonly model_get_seq_len: (a: number) => number;
-  readonly model_init_with_prompt: (a: number, b: number, c: number, d: number, e: number, f: number, g: number) => void;
+  readonly model_init_with_prompt: (a: number, b: number, c: number, d: number, e: number, f: number, g: number, h: number) => void;
   readonly model_next_token: (a: number, b: number) => void;
   readonly main: (a: number, b: number) => number;
   readonly __wbindgen_add_to_stack_pointer: (a: number) => number;

build/m.js

@@ -181,18 +181,19 @@ export class Model {
     /**
     * @param {string} prompt
     * @param {number} temp
+    * @param {number} top_p
     * @param {number} repeat_penalty
     * @param {bigint} seed
     * @returns {string}
     */
-    init_with_prompt(prompt, temp, repeat_penalty, seed) {
+    init_with_prompt(prompt, temp, top_p, repeat_penalty, seed) {
         let deferred3_0;
         let deferred3_1;
         try {
             const retptr = wasm.__wbindgen_add_to_stack_pointer(-16);
             const ptr0 = passStringToWasm0(prompt, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
             const len0 = WASM_VECTOR_LEN;
-            wasm.model_init_with_prompt(retptr, this.__wbg_ptr, ptr0, len0, temp, repeat_penalty, seed);
+            wasm.model_init_with_prompt(retptr, this.__wbg_ptr, ptr0, len0, temp, top_p, repeat_penalty, seed);
             var r0 = getInt32Memory0()[retptr / 4 + 0];
             var r1 = getInt32Memory0()[retptr / 4 + 1];
             var r2 = getInt32Memory0()[retptr / 4 + 2];

build/m_bg.wasm

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:702ada87fc9382641b032d5954eb68c05cb34fa1ffa25efe9699ee50cb745d8a
-size 3724190
+oid sha256:ff820d8875046447073f4079e83a6696f736572387d56d393092f995056e08b2
+size 3794192

build/m_bg.wasm.d.ts

@@ -4,7 +4,7 @@ export const memory: WebAssembly.Memory;
 export function __wbg_model_free(a: number): void;
 export function model_new(a: number, b: number, c: number, d: number, e: number): void;
 export function model_get_seq_len(a: number): number;
-export function model_init_with_prompt(a: number, b: number, c: number, d: number, e: number, f: number, g: number): void;
+export function model_init_with_prompt(a: number, b: number, c: number, d: number, e: number, f: number, g: number, h: number): void;
 export function model_next_token(a: number, b: number): void;
 export function main(a: number, b: number): number;
 export function __wbindgen_add_to_stack_pointer(a: number): number;