Sarah Ciston commited on
Commit
62c2dcb
·
1 Parent(s): 1bf613a
Files changed (1) hide show
  1. sketch.js +3 -2
sketch.js CHANGED
@@ -180,13 +180,14 @@ new p5(function(p5){
180
 
181
  async function runModel(PREPROMPT, PROMPT){
182
  // Chat completion API
183
- const out = await inference.chat_completion({ //inference.fill_mask({
184
  model: MODELNAME,
185
  // model: "google/gemma-2-9b",
186
  // messages: [{ role: "user", content: PREPROMPT + PROMPT }],
187
  max_tokens: 100
188
  });
189
-
 
190
  // let out = await pipe(PREPROMPT + PROMPT)
191
  // let out = await pipe(PREPROMPT + PROMPT, {
192
  // max_new_tokens: 250,
 
180
 
181
  async function runModel(PREPROMPT, PROMPT){
182
  // Chat completion API
183
+ const out = await inference.chat_completion({
184
  model: MODELNAME,
185
  // model: "google/gemma-2-9b",
186
  // messages: [{ role: "user", content: PREPROMPT + PROMPT }],
187
  max_tokens: 100
188
  });
189
+
190
+ //inference.fill_mask({
191
  // let out = await pipe(PREPROMPT + PROMPT)
192
  // let out = await pipe(PREPROMPT + PROMPT, {
193
  // max_new_tokens: 250,