diff options
Diffstat (limited to 'llm')
| -rw-r--r-- | llm/perplexity.go | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/llm/perplexity.go b/llm/perplexity.go index ab31889..5d9056e 100644 --- a/llm/perplexity.go +++ b/llm/perplexity.go @@ -1,7 +1,6 @@ package llm import ( - "context" "fmt" "sync" "time" @@ -47,16 +46,14 @@ func (e *Evaluator[R]) Run(data dataset.Reader, window, stride int) error { }() } - ctx, cancel := context.WithCancel(context.Background()) - - defer cancel() - pb := tui.NewProgressBar("Perplexity", 20, 0, time.Now()) - go pb.Watch(ctx, 1*time.Second, func() int { + pb.Start(1*time.Second, func() int { return int(e.completed.Load()) }) + defer pb.Close() + n := 0 for d := range data.Texts("text") { |