@@ -134,15 +134,15 @@ mod mmap {
134
134
// The components of the embedding matrix should be of type f32.
135
135
f32:: ensure_data_type ( read) ?;
136
136
137
- let n_padding = padding :: < f32 > ( read. seek ( SeekFrom :: Current ( 0 ) ) . map_err ( |e| {
137
+ let n_padding = padding :: < f32 > ( read. stream_position ( ) . map_err ( |e| {
138
138
Error :: read_error ( "Cannot get file position for computing padding" , e)
139
139
} ) ?) ;
140
140
read. seek ( SeekFrom :: Current ( n_padding as i64 ) )
141
141
. map_err ( |e| Error :: read_error ( "Cannot skip padding" , e) ) ?;
142
142
143
143
// Set up memory mapping.
144
144
let matrix_len = shape. size ( ) * size_of :: < f32 > ( ) ;
145
- let offset = read. seek ( SeekFrom :: Current ( 0 ) ) . map_err ( |e| {
145
+ let offset = read. stream_position ( ) . map_err ( |e| {
146
146
Error :: read_error (
147
147
"Cannot get file position for memory mapping embedding matrix" ,
148
148
e,
@@ -153,7 +153,7 @@ mod mmap {
153
153
mmap_opts
154
154
. offset ( offset)
155
155
. len ( matrix_len)
156
- . map ( & * read. get_ref ( ) )
156
+ . map ( read. get_ref ( ) )
157
157
. map_err ( |e| Error :: read_error ( "Cannot memory map embedding matrix" , e) ) ?
158
158
} ;
159
159
@@ -218,13 +218,13 @@ impl NdArray {
218
218
write
219
219
. write_u32 :: < LittleEndian > ( ChunkIdentifier :: NdArray as u32 )
220
220
. map_err ( |e| Error :: write_error ( "Cannot write embedding matrix chunk identifier" , e) ) ?;
221
- let n_padding = padding :: < f32 > ( write. seek ( SeekFrom :: Current ( 0 ) ) . map_err ( |e| {
221
+ let n_padding = padding :: < f32 > ( write. stream_position ( ) . map_err ( |e| {
222
222
Error :: write_error ( "Cannot get file position for computing padding" , e)
223
223
} ) ?) ;
224
224
225
225
let remaining_chunk_len = Self :: chunk_len (
226
226
data. view ( ) ,
227
- write. seek ( SeekFrom :: Current ( 0 ) ) . map_err ( |e| {
227
+ write. stream_position ( ) . map_err ( |e| {
228
228
Error :: read_error ( "Cannot get file position for computing padding" , e)
229
229
} ) ?,
230
230
) - ( size_of :: < u32 > ( ) + size_of :: < u64 > ( ) ) as u64 ;
@@ -346,7 +346,7 @@ impl ReadChunk for NdArray {
346
346
f32:: ensure_data_type ( read) ?;
347
347
348
348
let n_padding =
349
- padding :: < f32 > ( read. seek ( SeekFrom :: Current ( 0 ) ) . map_err ( |e| {
349
+ padding :: < f32 > ( read. stream_position ( ) . map_err ( |e| {
350
350
Error :: read_error ( "Cannot get file position for computing padding" , e)
351
351
} ) ?) ;
352
352
read. seek ( SeekFrom :: Current ( n_padding as i64 ) )
0 commit comments