try with repetition penalty and [mask]

sketch.js

@@ -17,8 +17,8 @@ async function textGenTask(pre, prompt){
   console.log('text-gen task initiated')
 
   // preprompt not working, fix later
-  // let INPUT = pre + prompt
-  let INPUT = prompt
+  let INPUT = pre + prompt
+  // let INPUT = prompt
 
   // PICK MODEL 
   let MODEL = 'Xenova/LaMini-Cerebras-256M'
@@ -27,9 +27,9 @@ async function textGenTask(pre, prompt){
   const pipe = await pipeline('text-generation', MODEL)
 
   // RUN INPUT THROUGH MODEL, 
-  var out = await pipe(INPUT, { max_new_tokens: 60, top_k: 90 })
+  var out = await pipe(INPUT, { max_new_tokens: 60, top_k: 90, repetition_penalty: 1.5 })
   // setting hyperparameters
-  // max_new_tokens: 256, top_k: 50, temperature: 0.7, do_sample: true,
+  // max_new_tokens: 256, top_k: 50, temperature: 0.7, do_sample: true, no_repeat_ngram_size: 2,
 
   console.log(await out)
   console.log('text-gen task completed')