Update to web-rwkv v0.2.3

Ai00-X · Aug 28, 2023 · 2f16b57 · 2f16b57
1 parent 454ed31
commit 2f16b57
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/main.rs b/src/main.rs
@@ -362,10 +362,7 @@ fn model_task(
         if embedding {
             let num_layer = model.info().num_layers;
             // let embedding = back.0[num_layer - 3][4 * num_emb..].to_vec();
-            let embedding = back
-                .state
-                .as_slice((.., 5 * (num_layer - 3) + 4, ..))?
-                .to_vec();
+            let embedding = back.as_slice((.., 5 * (num_layer - 3) + 4, ..))?.to_vec();
             let _ = token_sender.send(Token::Embed(embedding));
         }