-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathandroid_memory_fix.patch
More file actions
32 lines (27 loc) · 1.16 KB
/
android_memory_fix.patch
File metadata and controls
32 lines (27 loc) · 1.16 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
diff --git a/JOSIEApp-Android/app/src/main/java/com/josie/ai/LlamaNative.kt b/JOSIEApp-Android/app/src/main/java/com/josie/ai/LlamaNative.kt
--- a/JOSIEApp-Android/app/src/main/java/com/josie/ai/LlamaNative.kt
+++ b/JOSIEApp-Android/app/src/main/java/com/josie/ai/LlamaNative.kt
@@ -13,4 +13,5 @@
external fun loadModel(modelPath: String): Boolean
external fun generateStream(prompt: String, callback: StreamCallback)
+ external fun unload()
}
diff --git a/JOSIEApp-Android/app/src/main/java/com/josie/ai/JosieBrain.kt b/JOSIEApp-Android/app/src/main/java/com/josie/ai/JosieBrain.kt
--- a/JOSIEApp-Android/app/src/main/java/com/josie/ai/JosieBrain.kt
+++ b/JOSIEApp-Android/app/src/main/java/com/josie/ai/JosieBrain.kt
@@ -118,6 +118,7 @@
Log.d(TAG, "Loading model: $modelPath")
val success = withContext(Dispatchers.IO) {
+ llamaNative.unload()
llamaNative.loadModel(modelPath)
}
@@ -207,5 +208,10 @@
generationJob.join()
}
}
+
+ override fun onCleared() {
+ super.onCleared()
+ llamaNative.unload()
+ }
}