From c78ce765016392673805ed8dfafb4ae1a7b6c26f Mon Sep 17 00:00:00 2001 From: Laurent Mazare Date: Fri, 18 Aug 2023 09:38:22 +0100 Subject: Add a simple Module trait and implement it for the various nn layers (#500) * Start adding the module trait. * Use the module trait. * Implement module for qmatmul. --- candle-wasm-examples/llama2-c/src/model.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'candle-wasm-examples/llama2-c') diff --git a/candle-wasm-examples/llama2-c/src/model.rs b/candle-wasm-examples/llama2-c/src/model.rs index 2c867793..3fedb1d3 100644 --- a/candle-wasm-examples/llama2-c/src/model.rs +++ b/candle-wasm-examples/llama2-c/src/model.rs @@ -1,5 +1,5 @@ use candle::{DType, Device, IndexOp, Result, Tensor, D}; -use candle_nn::{rms_norm, Embedding, Linear, RmsNorm, VarBuilder}; +use candle_nn::{rms_norm, Embedding, Linear, Module, RmsNorm, VarBuilder}; use std::collections::HashMap; use std::sync::{Arc, Mutex}; -- cgit v1.2.3