sarahciston commited on
Commit
5a03e3a
·
verified ·
1 Parent(s): 8306ee8

move async call for async/await iterative model runs

Browse files
Files changed (1) hide show
  1. sketch.js +2 -2
sketch.js CHANGED
@@ -49,9 +49,9 @@ async function textGenTask(pre, prompt, blanks){
49
 
50
  var hyperparameters = { max_new_tokens: 60, top_k: 90, repetition_penalty: 1.5 }
51
 
52
- async promptArray.forEach(INPUT => {
53
  // RUN INPUT THROUGH MODEL,
54
- var out = await pipe(INPUT, hyperparameters)
55
  // setting hyperparameters
56
  // max_new_tokens: 256, top_k: 50, temperature: 0.7, do_sample: true, no_repeat_ngram_size: 2,
57
  // , num_return_sequences: 2 (must be 1?)
 
49
 
50
  var hyperparameters = { max_new_tokens: 60, top_k: 90, repetition_penalty: 1.5 }
51
 
52
+ promptArray.forEach(async i => {
53
  // RUN INPUT THROUGH MODEL,
54
+ var out = await pipe(i, hyperparameters)
55
  // setting hyperparameters
56
  // max_new_tokens: 256, top_k: 50, temperature: 0.7, do_sample: true, no_repeat_ngram_size: 2,
57
  // , num_return_sequences: 2 (must be 1?)