sarahciston commited on
Commit
76ed4e2
·
verified ·
1 Parent(s): dbaab50

remove chat template functions

Browse files
Files changed (1) hide show
  1. sketch.js +8 -8
sketch.js CHANGED
@@ -19,16 +19,16 @@ async function textGenTask(input){
19
 
20
  // const = modelsList = ['Xenova/LaMini-Cerebras-256M', ]
21
 
22
- let messages = [
23
- {"role": "system", "content": PREPROMPT},
24
- {"role": "user", "content": input}
25
- ]
26
 
27
- const prompt = pipe.tokenizer.apply_chat_template(messages, {
28
- tokenize: false, add_generation_prompt: true,
29
- });
30
 
31
- var out = await pipe(prompt, {
32
  max_new_tokens: 256,
33
  temperature: 0.7,
34
  do_sample: true,
 
19
 
20
  // const = modelsList = ['Xenova/LaMini-Cerebras-256M', ]
21
 
22
+ // let messages = [
23
+ // {"role": "system", "content": PREPROMPT},
24
+ // {"role": "user", "content": input}
25
+ // ]
26
 
27
+ // const prompt = pipe.tokenizer.apply_chat_template(messages, {
28
+ // tokenize: false, add_generation_prompt: true,
29
+ // });
30
 
31
+ var out = await pipe(input, {
32
  max_new_tokens: 256,
33
  temperature: 0.7,
34
  do_sample: true,