summaryrefslogtreecommitdiff
path: root/llm/causal.go
diff options
context:
space:
mode:
authorJonas Knobloch <jonas.knobloch@t-online.de>2026-04-04 22:14:02 +0200
committerJonas Knobloch <jonas.knobloch@t-online.de>2026-04-04 22:14:02 +0200
commit60b71d9f78895d78a641b1109e2b0ca293835698 (patch)
tree0057c3c35362e7b6489476be02e8e6a0edfa9619 /llm/causal.go
parenta96c1bf68bf5c477e1d3f105c18346c80d80b47a (diff)
Add score method
Diffstat (limited to 'llm/causal.go')
-rw-r--r--llm/causal.go1
1 files changed, 1 insertions, 0 deletions
diff --git a/llm/causal.go b/llm/causal.go
index 296a744..b683985 100644
--- a/llm/causal.go
+++ b/llm/causal.go
@@ -2,4 +2,5 @@ package llm
type Causal interface {
Generate(prompt []int64, steps int64, logits *[][]float32) ([]int64, error)
+ Score(tokens []int64, logProbs *[]float32) error
}