From ede63708cec2401d79a94ae1b36bd60f21e1d542 Mon Sep 17 00:00:00 2001
From: Xin Yang <xyang19@gmail.com>
Date: Thu, 30 May 2024 16:35:27 -0700
Subject: [PATCH] [rust] Fix deleteModel error

---
 extensions/tokenizers/rust/src/models/mod.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/extensions/tokenizers/rust/src/models/mod.rs b/extensions/tokenizers/rust/src/models/mod.rs
index a4db801bb52..b915f922982 100644
--- a/extensions/tokenizers/rust/src/models/mod.rs
+++ b/extensions/tokenizers/rust/src/models/mod.rs
@@ -124,7 +124,8 @@ pub extern "system" fn Java_ai_djl_engine_rust_RustLibrary_loadModel<'local>(
 
 #[no_mangle]
 pub extern "system" fn Java_ai_djl_engine_rust_RustLibrary_deleteModel<'local>(
-    _env: JObject,
+    _: JNIEnv,
+    _: JObject,
     handle: jlong,
 ) {
     drop_handle::<Box<dyn Model>>(handle);