diff options
Diffstat (limited to 'llm/perplexity.go')
| -rw-r--r-- | llm/perplexity.go | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/llm/perplexity.go b/llm/perplexity.go index 5d9056e..df48ff6 100644 --- a/llm/perplexity.go +++ b/llm/perplexity.go @@ -9,7 +9,7 @@ import ( "go.jknobloc.com/x/tui" ) -func (e *Evaluator[R]) Run(data dataset.Reader, window, stride int) error { +func (e *Evaluator[R]) Run(title string, data dataset.Reader, window, stride int) error { devices := make([]int, len(e.models)) for i := range len(devices) { @@ -46,7 +46,7 @@ func (e *Evaluator[R]) Run(data dataset.Reader, window, stride int) error { }() } - pb := tui.NewProgressBar("Perplexity", 20, 0, time.Now()) + pb := tui.NewProgressBar(title, 20, 0, time.Now()) pb.Start(1*time.Second, func() int { return int(e.completed.Load()) |