diff options
| author | Nicolas Patry <patry.nicolas@protonmail.com> | 2023-07-03 12:00:35 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-03 12:00:35 +0200 |
| commit | d0d530dfdce04d5fb656b10b4eb1bfd26dea37e8 (patch) | |
| tree | 0d12f1d27dbb0996def62dc316415b5f9a760c8a /candle-core/examples/llama/weights.rs | |
| parent | 48089005f6585c2f603f0d930a4a94b5aa18402f (diff) | |
| parent | 81cec86e758390e5f025a6e93888673b003fb4c8 (diff) | |
| download | candle-d0d530dfdce04d5fb656b10b4eb1bfd26dea37e8.tar.gz candle-d0d530dfdce04d5fb656b10b4eb1bfd26dea37e8.tar.bz2 candle-d0d530dfdce04d5fb656b10b4eb1bfd26dea37e8.zip | |
Merge pull request #59 from LaurentMazare/safety
Adding a bit more docs around safety.
Diffstat (limited to 'candle-core/examples/llama/weights.rs')
| -rw-r--r-- | candle-core/examples/llama/weights.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/examples/llama/weights.rs b/candle-core/examples/llama/weights.rs index cc3fccd4..c3364cef 100644 --- a/candle-core/examples/llama/weights.rs +++ b/candle-core/examples/llama/weights.rs @@ -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() |