FIXED TESTS

This commit is contained in:
Josh Yan
2024-07-16 12:14:10 -07:00
parent fc1b3ee9bf
commit fa49bfc0bd
3 changed files with 5 additions and 3 deletions

View File

@@ -260,7 +260,7 @@ func parseFromFile(ctx context.Context, file *os.File, digest string, fn func(ap
WriterTo: &llm.TensorWriter{
// This needs offset + tensors.Offset int64(tensor.Offset) to be correct
Reader: io.NewSectionReader(file, int64(tensor.Offset), int64(tensor.Size())),
Reader: io.NewSectionReader(file, offset + int64(tensor.Offset), int64(tensor.Size())),
},
})
}