lmz radames commited on
Commit
1dbcbfb
1 Parent(s): 66ff768

- fix (7aab05e6d84414909b031e6cb7884c2231a6a89c)


Co-authored-by: Radamés Ajna <radames@users.noreply.huggingface.co>

Files changed (4) hide show
  1. build/m.d.ts +3 -2
  2. build/m.js +3 -2
  3. build/m_bg.wasm +2 -2
  4. build/m_bg.wasm.d.ts +1 -1
build/m.d.ts CHANGED
@@ -16,11 +16,12 @@ export class Model {
16
  /**
17
  * @param {string} prompt
18
  * @param {number} temp
 
19
  * @param {number} repeat_penalty
20
  * @param {bigint} seed
21
  * @returns {string}
22
  */
23
- init_with_prompt(prompt: string, temp: number, repeat_penalty: number, seed: bigint): string;
24
  /**
25
  * @returns {string}
26
  */
@@ -34,7 +35,7 @@ export interface InitOutput {
34
  readonly __wbg_model_free: (a: number) => void;
35
  readonly model_new: (a: number, b: number, c: number, d: number, e: number) => void;
36
  readonly model_get_seq_len: (a: number) => number;
37
- readonly model_init_with_prompt: (a: number, b: number, c: number, d: number, e: number, f: number, g: number) => void;
38
  readonly model_next_token: (a: number, b: number) => void;
39
  readonly main: (a: number, b: number) => number;
40
  readonly __wbindgen_add_to_stack_pointer: (a: number) => number;
 
16
  /**
17
  * @param {string} prompt
18
  * @param {number} temp
19
+ * @param {number} top_p
20
  * @param {number} repeat_penalty
21
  * @param {bigint} seed
22
  * @returns {string}
23
  */
24
+ init_with_prompt(prompt: string, temp: number, top_p: number, repeat_penalty: number, seed: bigint): string;
25
  /**
26
  * @returns {string}
27
  */
 
35
  readonly __wbg_model_free: (a: number) => void;
36
  readonly model_new: (a: number, b: number, c: number, d: number, e: number) => void;
37
  readonly model_get_seq_len: (a: number) => number;
38
+ readonly model_init_with_prompt: (a: number, b: number, c: number, d: number, e: number, f: number, g: number, h: number) => void;
39
  readonly model_next_token: (a: number, b: number) => void;
40
  readonly main: (a: number, b: number) => number;
41
  readonly __wbindgen_add_to_stack_pointer: (a: number) => number;
build/m.js CHANGED
@@ -181,18 +181,19 @@ export class Model {
181
  /**
182
  * @param {string} prompt
183
  * @param {number} temp
 
184
  * @param {number} repeat_penalty
185
  * @param {bigint} seed
186
  * @returns {string}
187
  */
188
- init_with_prompt(prompt, temp, repeat_penalty, seed) {
189
  let deferred3_0;
190
  let deferred3_1;
191
  try {
192
  const retptr = wasm.__wbindgen_add_to_stack_pointer(-16);
193
  const ptr0 = passStringToWasm0(prompt, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
194
  const len0 = WASM_VECTOR_LEN;
195
- wasm.model_init_with_prompt(retptr, this.__wbg_ptr, ptr0, len0, temp, repeat_penalty, seed);
196
  var r0 = getInt32Memory0()[retptr / 4 + 0];
197
  var r1 = getInt32Memory0()[retptr / 4 + 1];
198
  var r2 = getInt32Memory0()[retptr / 4 + 2];
 
181
  /**
182
  * @param {string} prompt
183
  * @param {number} temp
184
+ * @param {number} top_p
185
  * @param {number} repeat_penalty
186
  * @param {bigint} seed
187
  * @returns {string}
188
  */
189
+ init_with_prompt(prompt, temp, top_p, repeat_penalty, seed) {
190
  let deferred3_0;
191
  let deferred3_1;
192
  try {
193
  const retptr = wasm.__wbindgen_add_to_stack_pointer(-16);
194
  const ptr0 = passStringToWasm0(prompt, wasm.__wbindgen_malloc, wasm.__wbindgen_realloc);
195
  const len0 = WASM_VECTOR_LEN;
196
+ wasm.model_init_with_prompt(retptr, this.__wbg_ptr, ptr0, len0, temp, top_p, repeat_penalty, seed);
197
  var r0 = getInt32Memory0()[retptr / 4 + 0];
198
  var r1 = getInt32Memory0()[retptr / 4 + 1];
199
  var r2 = getInt32Memory0()[retptr / 4 + 2];
build/m_bg.wasm CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:702ada87fc9382641b032d5954eb68c05cb34fa1ffa25efe9699ee50cb745d8a
3
- size 3724190
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff820d8875046447073f4079e83a6696f736572387d56d393092f995056e08b2
3
+ size 3794192
build/m_bg.wasm.d.ts CHANGED
@@ -4,7 +4,7 @@ export const memory: WebAssembly.Memory;
4
  export function __wbg_model_free(a: number): void;
5
  export function model_new(a: number, b: number, c: number, d: number, e: number): void;
6
  export function model_get_seq_len(a: number): number;
7
- export function model_init_with_prompt(a: number, b: number, c: number, d: number, e: number, f: number, g: number): void;
8
  export function model_next_token(a: number, b: number): void;
9
  export function main(a: number, b: number): number;
10
  export function __wbindgen_add_to_stack_pointer(a: number): number;
 
4
  export function __wbg_model_free(a: number): void;
5
  export function model_new(a: number, b: number, c: number, d: number, e: number): void;
6
  export function model_get_seq_len(a: number): number;
7
+ export function model_init_with_prompt(a: number, b: number, c: number, d: number, e: number, f: number, g: number, h: number): void;
8
  export function model_next_token(a: number, b: number): void;
9
  export function main(a: number, b: number): number;
10
  export function __wbindgen_add_to_stack_pointer(a: number): number;