huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-20 20:09:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Simple example fix.

Browse Source
This commit is contained in:
Ubuntu
2023-06-29 11:10:57 +00:00
parent 3872dc4751
commit 1913512f42
2 changed files with 58 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
candle-core/examples/llama/var_store.rs
View File

@ -1,7 +1,6 @@
use super::*;
use candle::{DType, Device, Result, Shape, Tensor, WithDType};
use std::collections::HashMap;
use std::path::PathBuf;
use std::sync::Arc;
#[allow(dead_code)]
@ -142,11 +141,11 @@ impl Block {
impl Llama {
pub fn load_npy(
device: &Device,
_filenames: &[PathBuf],
filename: &str,
cache: &Cache,
config: &Config,
) -> Result<Self> {
let weight_path = std::path::Path::new("/data/llama.npz");
let weight_path = std::path::Path::new(filename);
let weights = if weight_path.exists() {
println!("loading weights from {weight_path:?}");
let start_load = std::time::Instant::now();