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()); } ///