Adding a bit more docs around safety.

This commit is contained in:
Nicolas Patry
2023-07-03 11:55:54 +02:00
parent 48089005f6
commit 81cec86e75
2 changed files with 13 additions and 3 deletions

View File

@ -105,7 +105,7 @@ impl Llama {
) -> Result<Self> {
let handles: Vec<_> = filenames
.iter()
.map(candle::safetensors::MmapedFile::new)
.map(|f| unsafe { candle::safetensors::MmapedFile::new(f) })
.collect::<Result<Vec<_>>>()?;
let tensors: Vec<_> = handles
.iter()