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

llama2-c wasm fix.

Browse Source
This commit is contained in:
Laurent
2023-11-02 10:31:47 +01:00
parent fbd69f952c
commit b97463098c
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-wasm-examples/llama2-c/src/model.rs
View File

@ -1,5 +1,7 @@
use candle::{DType, Device, IndexOp, Result, Tensor, D};
use candle_nn::{embedding, linear, rms_norm, Embedding, Linear, Module, RmsNorm, VarBuilder};
use candle_nn::{
embedding, linear_no_bias as linear, rms_norm, Embedding, Linear, Module, RmsNorm, VarBuilder,
};
use std::collections::HashMap;
use std::sync::{Arc, Mutex};