summaryrefslogtreecommitdiff
path: root/llm/run.go
diff options
context:
space:
mode:
Diffstat (limited to 'llm/run.go')
-rw-r--r--llm/run.go2
1 files changed, 1 insertions, 1 deletions
diff --git a/llm/run.go b/llm/run.go
index 1843c94..483cd5d 100644
--- a/llm/run.go
+++ b/llm/run.go
@@ -63,7 +63,7 @@ func (e *Evaluator[R]) Run(title string, data dataset.Reader, window, stride int
b := newBatch(e.batchSize)
- for d := range data.Texts("text") {
+ for d := range data.Texts() {
for w, s := range tb.Push(n, d) {
if s == 0 {
s = 1 // first token as context