diff --git a/LLama/Abstractions/IModelParams.cs b/LLama/Abstractions/IModelParams.cs
index d25b3cf0..2ecfe49c 100644
--- a/LLama/Abstractions/IModelParams.cs
+++ b/LLama/Abstractions/IModelParams.cs
@@ -134,9 +134,10 @@ namespace LLama.Abstractions
///
public TensorSplitsCollection(float[] splits)
{
- if (splits.Length != Splits.Length)
- throw new ArgumentException($"tensor splits length must equal {Splits.Length}");
- Splits = splits;
+ if (splits.Length > Splits.Length)
+ throw new ArgumentException($"Must supply at most {Splits.Length} tensor splits", nameof(splits));
+
+ splits.CopyTo(Splits.AsSpan());
}
///