[rust] Fix deleteModel error

deepjavalibrary · May 30, 2024 · ec1c893 · ec1c893
1 parent ddbb95a
commit ec1c893
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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,
+    _env: JNIEnv,
+    _: JObject,
     handle: jlong,
 ) {
     drop_handle::<Box<dyn Model>>(handle);