Fix context leak.

Cargo.toml

@@ -9,7 +9,7 @@ keywords = ["deep-learning", "language", "model", "rwkv"]
 license = "MIT OR Apache-2.0"
 name = "web-rwkv"
 repository = "https://github.com/cryscan/web-rwkv"
-version = "0.6.30"
+version = "0.6.31"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 

src/context.rs

@@ -148,11 +148,15 @@ impl<'a> ContextBuilder {
         // start a thread for reading back buffers
         #[cfg(not(target_arch = "wasm32"))]
         {
-            let context = context.clone();
+            let context = Arc::downgrade(&context);
             std::thread::spawn(move || {
                 while let Ok((buffer, sender)) = receiver.recv() {
-                    let data = context.read_back_buffer(buffer);
-                    let _ = sender.send(data);
+                    if let Some(context) = context.upgrade() {
+                        let data = context.read_back_buffer(buffer);
+                        let _ = sender.send(data);
+                    } else {
+                        break;
+                    }
                 }
             });
         }
@@ -235,7 +239,7 @@ impl PartialEq for Context {
 
 impl Eq for Context {}
 
-impl Context {
+impl ContextInternal {
     pub fn checkout_pipeline(
         &self,
         name: impl AsRef<str>,