Method load_all

cake-core/src/utils/tensor_storage.rs:296–307

Load all tensors as a HashMap suitable for VarBuilder::from_tensors().

(
        &self,
        dtype: candle_core::DType,
        device: &candle_core::Device,
    )

    /// Load all tensors as a HashMap suitable for VarBuilder::from_tensors().
    pub fn load_all(
        &self,
        dtype: candle_core::DType,
        device: &candle_core::Device,
    ) -> Result<HashMap<String, candle_core::Tensor>> {
        let mut map = HashMap::with_capacity(self.index.len());
        for name in self.index.keys() {
            let tensor = self.load_tensor(name, dtype, device)?;
            map.insert(name.clone(), tensor);
        }
        Ok(map)
    }

    /// Build from a multi-shard model with index JSON.
    fn from_index_json(index_path: &Path) -> Result<Self> {