Add ArgMaxSampler public init (#392)

petrukha-ivan · web-flow · commit b3acd8d5360c · 2025-09-22T11:17:44.000-07:00
diff --git a/Libraries/MLXLMCommon/Evaluate.swift b/Libraries/MLXLMCommon/Evaluate.swift
@@ -126,7 +126,9 @@ public struct GenerateParameters: Sendable {
 
 /// Sampler that uses `argMax` (most likely) to sample the logits.
 public struct ArgMaxSampler: LogitSampler {
-    public func sample(logits: MLX.MLXArray) -> MLX.MLXArray {
+    public init() {}
+
+    public func sample(logits: MLXArray) -> MLXArray {
         argMax(logits, axis: -1)
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -126,7 +126,9 @@ public struct GenerateParameters: Sendable {`
`126`	`126`
`127`	`127`	/// Sampler that uses `argMax` (most likely) to sample the logits.
`128`	`128`	`public struct ArgMaxSampler: LogitSampler {`
`129`		`- public func sample(logits: MLX.MLXArray) -> MLX.MLXArray {`
	`129`	`+ public init() {}`
	`130`	`+`
	`131`	`+ public func sample(logits: MLXArray) -> MLXArray {`
`130`	`132`	`argMax(logits, axis: -1)`
`131`	`133`	`}`
`132`	`134`	`}`