distillgpt2 model

sketch.js

@@ -20,22 +20,15 @@ async function textGenTask(pre, prompt){
   let INPUT = prompt
 
   // PICK MODEL 
-  let MODEL = 'Xenova/bloom-560m'
+  let MODEL = 'Xenova/distilgpt2'
   // const = modelsList = ['Xenova/LaMini-Cerebras-256M', 'Xenova/TinyLlama-1.1B-Chat-v1.0', 'Xenova/distilgpt2', 'Xenova/bloom-560m']
 
   const pipe = await pipeline('text-generation', MODEL)
 
   // RUN INPUT THROUGH MODEL, 
-  var out = await pipe(INPUT)
-
-  /* //setting hyperparameters
-    , {
-      max_new_tokens: 256,
-      top_k: 50,
-      temperature: 0.7,
-      do_sample: true,
-    })
-  */
+  var out = await pipe(INPUT, { top_k: 30, max_new_tokens: 100 })
+  // setting hyperparameters
+  // max_new_tokens: 256, top_k: 50, temperature: 0.7, do_sample: true,
 
   console.log(await out)
   console.log('text-gen task completed')
@@ -158,8 +151,8 @@ new p5(function (p5){
     let outs = await textGenTask(PREPROMPT, PROMPT_INPUT)
     console.log(outs)
   
-    let outText = p5.createP('')
-    await outText.html(outs) // true appends text instead of replaces
+    let outText = p5.createP('...')
+    await outText.html(outs, 'false') // true appends text instead of replaces
   }
 
 });