huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Depreate the VarBuilder::from_safetensors function. (#951)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-24 11:18:17 +01:00
committed by GitHub
parent bcb0ed8f1c
commit 4aeb449017
2 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
candle-nn/src/var_builder.rs
View File

@ -1,6 +1,6 @@
//! A `VarBuilder` is used to retrieve variables used by a model. These variables can either come
//! from a pre-trained checkpoint, e.g. using `VarBuilder::from_safetensors`, or initialized for
//! training, e.g. using `VarBuilder::from_varmap`.
//! from a pre-trained checkpoint, e.g. using `VarBuilder::from_mmaped_safetensors`, or initialized
//! for training, e.g. using `VarBuilder::from_varmap`.
use crate::VarMap;
use candle::{safetensors::Load, DType, Device, Error, Result, Shape, Tensor};
use safetensors::{slice::IndexOp, tensor::SafeTensors};
@ -414,6 +414,10 @@ impl<'a> VarBuilder<'a> {
/// Initializes a `VarBuilder` that retrieves tensors stored in a collection of safetensors
/// data.
#[deprecated(
since = "0.2.3",
note = "use from_mmaped_safetensors or from_buffered_safetensors instead"
)]
pub fn from_safetensors(safetensors: Vec<SafeTensors<'a>>, dtype: DType, dev: &Device) -> Self {
let mut routing = HashMap::new();
for (index, sf) in safetensors.iter().enumerate() {