cui-llama.rn 1.5.0 → 1.6.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (309) hide show
  1. package/LICENSE +20 -20
  2. package/README.md +317 -319
  3. package/android/build.gradle +116 -116
  4. package/android/gradle.properties +5 -5
  5. package/android/src/main/AndroidManifest.xml +4 -4
  6. package/android/src/main/CMakeLists.txt +124 -124
  7. package/android/src/main/java/com/rnllama/LlamaContext.java +645 -645
  8. package/android/src/main/java/com/rnllama/RNLlama.java +695 -695
  9. package/android/src/main/java/com/rnllama/RNLlamaPackage.java +48 -48
  10. package/android/src/main/jni-utils.h +100 -100
  11. package/android/src/main/jni.cpp +1263 -1263
  12. package/android/src/main/jniLibs/arm64-v8a/librnllama.so +0 -0
  13. package/android/src/main/jniLibs/arm64-v8a/librnllama_v8.so +0 -0
  14. package/android/src/main/jniLibs/arm64-v8a/librnllama_v8_2.so +0 -0
  15. package/android/src/main/jniLibs/arm64-v8a/librnllama_v8_2_dotprod.so +0 -0
  16. package/android/src/main/jniLibs/arm64-v8a/librnllama_v8_2_dotprod_i8mm.so +0 -0
  17. package/android/src/main/jniLibs/arm64-v8a/librnllama_v8_2_i8mm.so +0 -0
  18. package/android/src/main/jniLibs/x86_64/librnllama.so +0 -0
  19. package/android/src/main/jniLibs/x86_64/librnllama_x86_64.so +0 -0
  20. package/android/src/newarch/java/com/rnllama/RNLlamaModule.java +135 -135
  21. package/android/src/oldarch/java/com/rnllama/RNLlamaModule.java +136 -136
  22. package/cpp/README.md +4 -4
  23. package/cpp/ggml-llama-sim.metallib +0 -0
  24. package/cpp/ggml-llama.metallib +0 -0
  25. package/cpp/ggml-metal-impl.h +597 -597
  26. package/cpp/ggml-metal.m +4 -0
  27. package/cpp/ggml.h +1 -1
  28. package/cpp/rn-llama.cpp +873 -873
  29. package/cpp/rn-llama.h +138 -138
  30. package/cpp/sampling.h +107 -107
  31. package/cpp/unicode-data.cpp +7034 -7034
  32. package/cpp/unicode-data.h +20 -20
  33. package/cpp/unicode.cpp +849 -849
  34. package/cpp/unicode.h +66 -66
  35. package/ios/CMakeLists.txt +116 -108
  36. package/ios/RNLlama.h +7 -7
  37. package/ios/RNLlama.mm +418 -405
  38. package/ios/RNLlamaContext.h +57 -57
  39. package/ios/RNLlamaContext.mm +835 -835
  40. package/ios/rnllama.xcframework/Info.plist +74 -74
  41. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/binary-ops.h +16 -0
  42. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/chat.h +143 -0
  43. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/common.h +677 -0
  44. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/cpu-common.h +72 -0
  45. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-alloc.h +76 -0
  46. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-backend-impl.h +255 -0
  47. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-backend.h +354 -0
  48. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-common.h +1857 -0
  49. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-cpp.h +39 -0
  50. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-cpu-aarch64.h +8 -0
  51. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-cpu-impl.h +512 -0
  52. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-cpu-quants.h +63 -0
  53. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-cpu-traits.h +38 -0
  54. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-cpu.h +138 -0
  55. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-impl.h +594 -0
  56. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-metal-impl.h +597 -0
  57. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-metal.h +66 -0
  58. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-opt.h +216 -0
  59. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-quants.h +100 -0
  60. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml-threading.h +14 -0
  61. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ggml.h +2222 -0
  62. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/gguf.h +202 -0
  63. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/json-schema-to-grammar.h +21 -0
  64. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/json.hpp +24766 -0
  65. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-adapter.h +76 -0
  66. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-arch.h +428 -0
  67. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-batch.h +88 -0
  68. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-chat.h +56 -0
  69. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-context.h +265 -0
  70. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-cparams.h +38 -0
  71. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-cpp.h +30 -0
  72. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-grammar.h +173 -0
  73. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-graph.h +592 -0
  74. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-hparams.h +156 -0
  75. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-impl.h +61 -0
  76. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-io.h +35 -0
  77. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-kv-cache.h +213 -0
  78. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-memory.h +21 -0
  79. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-mmap.h +68 -0
  80. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-model-loader.h +169 -0
  81. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-model.h +409 -0
  82. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-sampling.h +32 -0
  83. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama-vocab.h +125 -0
  84. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/llama.h +1434 -0
  85. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/log.h +132 -0
  86. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/minja/chat-template.hpp +537 -0
  87. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/minja/minja.hpp +2941 -0
  88. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/ops.h +128 -0
  89. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/rn-llama.h +138 -0
  90. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/sampling.h +107 -0
  91. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/sgemm.h +14 -0
  92. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/simd-mappings.h +888 -0
  93. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/speculative.h +28 -0
  94. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/unary-ops.h +28 -0
  95. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/unicode-data.h +20 -0
  96. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/unicode.h +66 -0
  97. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Headers/vec.h +802 -0
  98. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/Info.plist +0 -0
  99. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/ggml-llama.metallib +0 -0
  100. package/ios/rnllama.xcframework/ios-arm64/rnllama.framework/rnllama +0 -0
  101. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/binary-ops.h +16 -0
  102. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/chat.h +143 -0
  103. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/common.h +677 -0
  104. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/cpu-common.h +72 -0
  105. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-alloc.h +76 -0
  106. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-backend-impl.h +255 -0
  107. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-backend.h +354 -0
  108. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-common.h +1857 -0
  109. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-cpp.h +39 -0
  110. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-cpu-aarch64.h +8 -0
  111. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-cpu-impl.h +512 -0
  112. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-cpu-quants.h +63 -0
  113. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-cpu-traits.h +38 -0
  114. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-cpu.h +138 -0
  115. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-impl.h +594 -0
  116. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-metal-impl.h +597 -0
  117. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-metal.h +66 -0
  118. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-opt.h +216 -0
  119. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-quants.h +100 -0
  120. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-threading.h +14 -0
  121. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ggml.h +2222 -0
  122. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/gguf.h +202 -0
  123. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/json-schema-to-grammar.h +21 -0
  124. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/json.hpp +24766 -0
  125. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-adapter.h +76 -0
  126. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-arch.h +428 -0
  127. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-batch.h +88 -0
  128. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-chat.h +56 -0
  129. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-context.h +265 -0
  130. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-cparams.h +38 -0
  131. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-cpp.h +30 -0
  132. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-grammar.h +173 -0
  133. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-graph.h +592 -0
  134. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-hparams.h +156 -0
  135. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-impl.h +61 -0
  136. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-io.h +35 -0
  137. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-kv-cache.h +213 -0
  138. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-memory.h +21 -0
  139. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-mmap.h +68 -0
  140. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-model-loader.h +169 -0
  141. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-model.h +409 -0
  142. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-sampling.h +32 -0
  143. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama-vocab.h +125 -0
  144. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/llama.h +1434 -0
  145. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/log.h +132 -0
  146. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/minja/chat-template.hpp +537 -0
  147. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/minja/minja.hpp +2941 -0
  148. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/ops.h +128 -0
  149. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/rn-llama.h +138 -0
  150. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/sampling.h +107 -0
  151. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/sgemm.h +14 -0
  152. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/simd-mappings.h +888 -0
  153. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/speculative.h +28 -0
  154. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/unary-ops.h +28 -0
  155. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/unicode-data.h +20 -0
  156. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/unicode.h +66 -0
  157. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Headers/vec.h +802 -0
  158. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/Info.plist +0 -0
  159. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/_CodeSignature/CodeResources +101 -0
  160. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/ggml-llama-sim.metallib +0 -0
  161. package/ios/rnllama.xcframework/ios-arm64_x86_64-simulator/rnllama.framework/rnllama +0 -0
  162. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/binary-ops.h +16 -0
  163. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/chat.h +143 -0
  164. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/common.h +677 -0
  165. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/cpu-common.h +72 -0
  166. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-alloc.h +76 -0
  167. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-backend-impl.h +255 -0
  168. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-backend.h +354 -0
  169. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-common.h +1857 -0
  170. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-cpp.h +39 -0
  171. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-cpu-aarch64.h +8 -0
  172. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-cpu-impl.h +512 -0
  173. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-cpu-quants.h +63 -0
  174. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-cpu-traits.h +38 -0
  175. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-cpu.h +138 -0
  176. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-impl.h +594 -0
  177. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-metal-impl.h +597 -0
  178. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-metal.h +66 -0
  179. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-opt.h +216 -0
  180. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-quants.h +100 -0
  181. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml-threading.h +14 -0
  182. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ggml.h +2222 -0
  183. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/gguf.h +202 -0
  184. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/json-schema-to-grammar.h +21 -0
  185. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/json.hpp +24766 -0
  186. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-adapter.h +76 -0
  187. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-arch.h +428 -0
  188. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-batch.h +88 -0
  189. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-chat.h +56 -0
  190. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-context.h +265 -0
  191. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-cparams.h +38 -0
  192. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-cpp.h +30 -0
  193. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-grammar.h +173 -0
  194. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-graph.h +592 -0
  195. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-hparams.h +156 -0
  196. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-impl.h +61 -0
  197. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-io.h +35 -0
  198. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-kv-cache.h +213 -0
  199. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-memory.h +21 -0
  200. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-mmap.h +68 -0
  201. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-model-loader.h +169 -0
  202. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-model.h +409 -0
  203. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-sampling.h +32 -0
  204. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama-vocab.h +125 -0
  205. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/llama.h +1434 -0
  206. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/log.h +132 -0
  207. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/minja/chat-template.hpp +537 -0
  208. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/minja/minja.hpp +2941 -0
  209. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/ops.h +128 -0
  210. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/rn-llama.h +138 -0
  211. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/sampling.h +107 -0
  212. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/sgemm.h +14 -0
  213. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/simd-mappings.h +888 -0
  214. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/speculative.h +28 -0
  215. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/unary-ops.h +28 -0
  216. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/unicode-data.h +20 -0
  217. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/unicode.h +66 -0
  218. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Headers/vec.h +802 -0
  219. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/Info.plist +0 -0
  220. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/ggml-llama.metallib +0 -0
  221. package/ios/rnllama.xcframework/tvos-arm64/rnllama.framework/rnllama +0 -0
  222. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/binary-ops.h +16 -0
  223. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/chat.h +143 -0
  224. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/common.h +677 -0
  225. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/cpu-common.h +72 -0
  226. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-alloc.h +76 -0
  227. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-backend-impl.h +255 -0
  228. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-backend.h +354 -0
  229. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-common.h +1857 -0
  230. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-cpp.h +39 -0
  231. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-cpu-aarch64.h +8 -0
  232. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-cpu-impl.h +512 -0
  233. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-cpu-quants.h +63 -0
  234. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-cpu-traits.h +38 -0
  235. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-cpu.h +138 -0
  236. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-impl.h +594 -0
  237. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-metal-impl.h +597 -0
  238. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-metal.h +66 -0
  239. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-opt.h +216 -0
  240. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-quants.h +100 -0
  241. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml-threading.h +14 -0
  242. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ggml.h +2222 -0
  243. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/gguf.h +202 -0
  244. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/json-schema-to-grammar.h +21 -0
  245. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/json.hpp +24766 -0
  246. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-adapter.h +76 -0
  247. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-arch.h +428 -0
  248. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-batch.h +88 -0
  249. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-chat.h +56 -0
  250. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-context.h +265 -0
  251. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-cparams.h +38 -0
  252. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-cpp.h +30 -0
  253. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-grammar.h +173 -0
  254. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-graph.h +592 -0
  255. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-hparams.h +156 -0
  256. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-impl.h +61 -0
  257. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-io.h +35 -0
  258. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-kv-cache.h +213 -0
  259. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-memory.h +21 -0
  260. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-mmap.h +68 -0
  261. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-model-loader.h +169 -0
  262. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-model.h +409 -0
  263. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-sampling.h +32 -0
  264. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama-vocab.h +125 -0
  265. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/llama.h +1434 -0
  266. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/log.h +132 -0
  267. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/minja/chat-template.hpp +537 -0
  268. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/minja/minja.hpp +2941 -0
  269. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/ops.h +128 -0
  270. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/rn-llama.h +138 -0
  271. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/sampling.h +107 -0
  272. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/sgemm.h +14 -0
  273. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/simd-mappings.h +888 -0
  274. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/speculative.h +28 -0
  275. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/unary-ops.h +28 -0
  276. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/unicode-data.h +20 -0
  277. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/unicode.h +66 -0
  278. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Headers/vec.h +802 -0
  279. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/Info.plist +0 -0
  280. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/_CodeSignature/CodeResources +101 -0
  281. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/ggml-llama-sim.metallib +0 -0
  282. package/ios/rnllama.xcframework/tvos-arm64_x86_64-simulator/rnllama.framework/rnllama +0 -0
  283. package/jest/mock.js +203 -203
  284. package/lib/commonjs/NativeRNLlama.js +1 -2
  285. package/lib/commonjs/NativeRNLlama.js.map +1 -1
  286. package/lib/commonjs/chat.js.map +1 -1
  287. package/lib/commonjs/grammar.js +12 -31
  288. package/lib/commonjs/grammar.js.map +1 -1
  289. package/lib/commonjs/index.js +47 -47
  290. package/lib/commonjs/index.js.map +1 -1
  291. package/lib/commonjs/package.json +1 -0
  292. package/lib/module/NativeRNLlama.js +2 -0
  293. package/lib/module/NativeRNLlama.js.map +1 -1
  294. package/lib/module/chat.js +2 -0
  295. package/lib/module/chat.js.map +1 -1
  296. package/lib/module/grammar.js +14 -31
  297. package/lib/module/grammar.js.map +1 -1
  298. package/lib/module/index.js +47 -45
  299. package/lib/module/index.js.map +1 -1
  300. package/lib/module/package.json +1 -0
  301. package/lib/typescript/NativeRNLlama.d.ts +6 -4
  302. package/lib/typescript/NativeRNLlama.d.ts.map +1 -1
  303. package/lib/typescript/index.d.ts.map +1 -1
  304. package/llama-rn.podspec +48 -48
  305. package/package.json +233 -233
  306. package/src/NativeRNLlama.ts +426 -426
  307. package/src/chat.ts +44 -44
  308. package/src/grammar.ts +854 -854
  309. package/src/index.ts +495 -487
@@ -0,0 +1,677 @@
1
+ // Various helper functions and utilities
2
+
3
+ #pragma once
4
+
5
+ #include "llama-cpp.h"
6
+
7
+ #include <set>
8
+ #include <string>
9
+ #include <vector>
10
+ #include <sstream>
11
+
12
+ #ifdef _WIN32
13
+ #define DIRECTORY_SEPARATOR '\\'
14
+ #else
15
+ #define DIRECTORY_SEPARATOR '/'
16
+ #endif // _WIN32
17
+
18
+ #define die(msg) do { fputs("error: " msg "\n", stderr); exit(1); } while (0)
19
+ #define die_fmt(fmt, ...) do { fprintf(stderr, "error: " fmt "\n", __VA_ARGS__); exit(1); } while (0)
20
+
21
+ #define print_build_info() do { \
22
+ fprintf(stderr, "%s: build = %d (%s)\n", __func__, LLAMA_BUILD_NUMBER, LLAMA_COMMIT); \
23
+ fprintf(stderr, "%s: built with %s for %s\n", __func__, LLAMA_COMPILER, LLAMA_BUILD_TARGET); \
24
+ } while(0)
25
+
26
+ #define DEFAULT_MODEL_PATH "models/7B/ggml-model-f16.gguf"
27
+
28
+ struct common_adapter_lora_info {
29
+ std::string path;
30
+ float scale;
31
+
32
+ struct llama_adapter_lora * ptr;
33
+ };
34
+
35
+ using llama_tokens = std::vector<llama_token>;
36
+
37
+ // build info
38
+ extern int LLAMA_BUILD_NUMBER;
39
+ extern const char * LLAMA_COMMIT;
40
+ extern const char * LLAMA_COMPILER;
41
+ extern const char * LLAMA_BUILD_TARGET;
42
+
43
+ struct common_control_vector_load_info;
44
+
45
+ #define print_build_info() do { \
46
+ fprintf(stderr, "%s: build = %d (%s)\n", __func__, LLAMA_BUILD_NUMBER, LLAMA_COMMIT); \
47
+ fprintf(stderr, "%s: built with %s for %s\n", __func__, LLAMA_COMPILER, LLAMA_BUILD_TARGET); \
48
+ } while(0)
49
+
50
+ // build info
51
+ extern int LLAMA_BUILD_NUMBER;
52
+ extern char const *LLAMA_COMMIT;
53
+ extern char const *LLAMA_COMPILER;
54
+ extern char const *LLAMA_BUILD_TARGET;
55
+
56
+ //
57
+ // CPU utils
58
+ //
59
+
60
+ struct cpu_params {
61
+ int n_threads = -1;
62
+ bool cpumask[LM_GGML_MAX_N_THREADS] = {false}; // CPU affinity mask.
63
+ bool mask_valid = false; // Default: any CPU
64
+ enum lm_ggml_sched_priority priority = LM_GGML_SCHED_PRIO_NORMAL; // Scheduling prio : (0 - normal, 1 - medium, 2 - high, 3 - realtime)
65
+ bool strict_cpu = false; // Use strict CPU placement
66
+ uint32_t poll = 50; // Polling (busywait) level (0 - no polling, 100 - mostly polling)
67
+ };
68
+
69
+ int32_t cpu_get_num_physical_cores();
70
+ int32_t cpu_get_num_math();
71
+
72
+ //
73
+ // Common params
74
+ //
75
+
76
+ enum llama_example {
77
+ LLAMA_EXAMPLE_COMMON,
78
+ LLAMA_EXAMPLE_SPECULATIVE,
79
+ LLAMA_EXAMPLE_MAIN,
80
+ LLAMA_EXAMPLE_INFILL,
81
+ LLAMA_EXAMPLE_EMBEDDING,
82
+ LLAMA_EXAMPLE_PERPLEXITY,
83
+ LLAMA_EXAMPLE_RETRIEVAL,
84
+ LLAMA_EXAMPLE_PASSKEY,
85
+ LLAMA_EXAMPLE_IMATRIX,
86
+ LLAMA_EXAMPLE_BENCH,
87
+ LLAMA_EXAMPLE_SERVER,
88
+ LLAMA_EXAMPLE_CVECTOR_GENERATOR,
89
+ LLAMA_EXAMPLE_EXPORT_LORA,
90
+ LLAMA_EXAMPLE_LLAVA,
91
+ LLAMA_EXAMPLE_LOOKUP,
92
+ LLAMA_EXAMPLE_PARALLEL,
93
+ LLAMA_EXAMPLE_TTS,
94
+
95
+ LLAMA_EXAMPLE_COUNT,
96
+ };
97
+
98
+ enum common_sampler_type {
99
+ COMMON_SAMPLER_TYPE_NONE = 0,
100
+ COMMON_SAMPLER_TYPE_DRY = 1,
101
+ COMMON_SAMPLER_TYPE_TOP_K = 2,
102
+ COMMON_SAMPLER_TYPE_TOP_P = 3,
103
+ COMMON_SAMPLER_TYPE_MIN_P = 4,
104
+ //COMMON_SAMPLER_TYPE_TFS_Z = 5,
105
+ COMMON_SAMPLER_TYPE_TYPICAL_P = 6,
106
+ COMMON_SAMPLER_TYPE_TEMPERATURE = 7,
107
+ COMMON_SAMPLER_TYPE_XTC = 8,
108
+ COMMON_SAMPLER_TYPE_INFILL = 9,
109
+ COMMON_SAMPLER_TYPE_PENALTIES = 10,
110
+ };
111
+
112
+ // dimensionality reduction methods, used by cvector-generator
113
+ enum dimre_method {
114
+ DIMRE_METHOD_PCA,
115
+ DIMRE_METHOD_MEAN,
116
+ };
117
+
118
+ enum common_conversation_mode {
119
+ COMMON_CONVERSATION_MODE_DISABLED = 0,
120
+ COMMON_CONVERSATION_MODE_ENABLED = 1,
121
+ COMMON_CONVERSATION_MODE_AUTO = 2,
122
+ };
123
+
124
+ enum common_grammar_trigger_type {
125
+ COMMON_GRAMMAR_TRIGGER_TYPE_TOKEN,
126
+ COMMON_GRAMMAR_TRIGGER_TYPE_WORD,
127
+ COMMON_GRAMMAR_TRIGGER_TYPE_PATTERN,
128
+ COMMON_GRAMMAR_TRIGGER_TYPE_PATTERN_START,
129
+ };
130
+
131
+ struct common_grammar_trigger {
132
+ common_grammar_trigger_type type;
133
+ std::string value;
134
+ llama_token token = LLAMA_TOKEN_NULL;
135
+ };
136
+
137
+ // sampling parameters
138
+ struct common_params_sampling {
139
+ uint32_t seed = LLAMA_DEFAULT_SEED; // the seed used to initialize llama_sampler
140
+
141
+ int32_t n_prev = 64; // number of previous tokens to remember
142
+ int32_t n_probs = 0; // if greater than 0, output the probabilities of top n_probs tokens.
143
+ int32_t min_keep = 0; // 0 = disabled, otherwise samplers should return at least min_keep tokens
144
+ int32_t top_k = 40; // <= 0 to use vocab size
145
+ float top_p = 0.95f; // 1.0 = disabled
146
+ float min_p = 0.05f; // 0.0 = disabled
147
+ float xtc_probability = 0.00f; // 0.0 = disabled
148
+ float xtc_threshold = 0.10f; // > 0.5 disables XTC
149
+ float typ_p = 1.00f; // typical_p, 1.0 = disabled
150
+ float temp = 0.80f; // <= 0.0 to sample greedily, 0.0 to not output probabilities
151
+ float dynatemp_range = 0.00f; // 0.0 = disabled
152
+ float dynatemp_exponent = 1.00f; // controls how entropy maps to temperature in dynamic temperature sampler
153
+ int32_t penalty_last_n = 64; // last n tokens to penalize (0 = disable penalty, -1 = context size)
154
+ float penalty_repeat = 1.00f; // 1.0 = disabled
155
+ float penalty_freq = 0.00f; // 0.0 = disabled
156
+ float penalty_present = 0.00f; // 0.0 = disabled
157
+ float dry_multiplier = 0.0f; // 0.0 = disabled; DRY repetition penalty for tokens extending repetition:
158
+ float dry_base = 1.75f; // 0.0 = disabled; multiplier * base ^ (length of sequence before token - allowed length)
159
+ int32_t dry_allowed_length = 2; // tokens extending repetitions beyond this receive penalty
160
+ int32_t dry_penalty_last_n = -1; // how many tokens to scan for repetitions (0 = disable penalty, -1 = context size)
161
+ int32_t mirostat = 0; // 0 = disabled, 1 = mirostat, 2 = mirostat 2.0
162
+ float top_n_sigma = -1.00f;// -1.0 = disabled
163
+ float mirostat_tau = 5.00f; // target entropy
164
+ float mirostat_eta = 0.10f; // learning rate
165
+ bool ignore_eos = false;
166
+ bool no_perf = false; // disable performance metrics
167
+ bool timing_per_token = false;
168
+
169
+ std::vector<std::string> dry_sequence_breakers = {"\n", ":", "\"", "*"}; // default sequence breakers for DRY
170
+
171
+
172
+ std::vector<enum common_sampler_type> samplers = {
173
+ COMMON_SAMPLER_TYPE_PENALTIES,
174
+ COMMON_SAMPLER_TYPE_DRY,
175
+ COMMON_SAMPLER_TYPE_TOP_K,
176
+ COMMON_SAMPLER_TYPE_TYPICAL_P,
177
+ COMMON_SAMPLER_TYPE_TOP_P,
178
+ COMMON_SAMPLER_TYPE_MIN_P,
179
+ COMMON_SAMPLER_TYPE_XTC,
180
+ COMMON_SAMPLER_TYPE_TEMPERATURE,
181
+ };
182
+
183
+ std::string grammar; // optional BNF-like grammar to constrain sampling
184
+ bool grammar_lazy = false;
185
+ std::vector<common_grammar_trigger> grammar_triggers; // optional triggers (for lazy grammars)
186
+ std::set<llama_token> preserved_tokens;
187
+
188
+ std::vector<llama_logit_bias> logit_bias; // logit biases to apply
189
+
190
+ // print the parameters into a string
191
+ std::string print() const;
192
+ };
193
+
194
+ struct common_params_model {
195
+ std::string path = ""; // model local path // NOLINT
196
+ std::string url = ""; // model url to download // NOLINT
197
+ std::string hf_repo = ""; // HF repo // NOLINT
198
+ std::string hf_file = ""; // HF file // NOLINT
199
+ };
200
+
201
+ struct common_params_speculative {
202
+ std::vector<lm_ggml_backend_dev_t> devices; // devices to use for offloading
203
+
204
+ int32_t n_ctx = 0; // draft context size
205
+ int32_t n_max = 16; // maximum number of tokens to draft during speculative decoding
206
+ int32_t n_min = 0; // minimum number of draft tokens to use for speculative decoding
207
+ int32_t n_gpu_layers = -1; // number of layers to store in VRAM for the draft model (-1 - use default)
208
+ float p_split = 0.1f; // speculative decoding split probability
209
+ float p_min = 0.75f; // minimum speculative decoding probability (greedy)
210
+
211
+ struct cpu_params cpuparams;
212
+ struct cpu_params cpuparams_batch;
213
+
214
+ struct common_params_model model;
215
+ };
216
+
217
+ struct common_params_vocoder {
218
+ struct common_params_model model;
219
+
220
+ std::string speaker_file = ""; // speaker file path // NOLINT
221
+
222
+ bool use_guide_tokens = false; // enable guide tokens to improve TTS accuracy // NOLINT
223
+ };
224
+
225
+ enum common_reasoning_format {
226
+ COMMON_REASONING_FORMAT_NONE,
227
+ COMMON_REASONING_FORMAT_DEEPSEEK, // Extract thinking tag contents and return as `message.reasoning_content`
228
+ };
229
+
230
+ struct common_params {
231
+
232
+ void * progress_callback_user_data = nullptr;
233
+ llama_progress_callback progress_callback = nullptr;
234
+ bool vocab_only = false;
235
+ int32_t n_predict = -1; // new tokens to predict
236
+ int32_t n_ctx = 4096; // context size
237
+ int32_t n_batch = 2048; // logical batch size for prompt processing (must be >=32 to use BLAS)
238
+ int32_t n_ubatch = 512; // physical batch size for prompt processing (must be >=32 to use BLAS)
239
+ int32_t n_keep = 0; // number of tokens to keep from initial prompt
240
+ int32_t n_chunks = -1; // max number of chunks to process (-1 = unlimited)
241
+ int32_t n_parallel = 1; // number of parallel sequences to decode
242
+ int32_t n_sequences = 1; // number of sequences to decode
243
+ int32_t grp_attn_n = 1; // group-attention factor
244
+ int32_t grp_attn_w = 512; // group-attention width
245
+ int32_t n_print = -1; // print token count every n tokens (-1 = disabled)
246
+ float rope_freq_base = 0.0f; // RoPE base frequency
247
+ float rope_freq_scale = 0.0f; // RoPE frequency scaling factor
248
+ float yarn_ext_factor = -1.0f; // YaRN extrapolation mix factor
249
+ float yarn_attn_factor = 1.0f; // YaRN magnitude scaling factor
250
+ float yarn_beta_fast = 32.0f; // YaRN low correction dim
251
+ float yarn_beta_slow = 1.0f; // YaRN high correction dim
252
+ int32_t yarn_orig_ctx = 0; // YaRN original context length
253
+ float defrag_thold = 0.1f; // KV cache defragmentation threshold
254
+
255
+ // offload params
256
+ std::vector<lm_ggml_backend_dev_t> devices; // devices to use for offloading
257
+
258
+ int32_t n_gpu_layers = -1; // number of layers to store in VRAM (-1 - use default)
259
+ int32_t main_gpu = 0; // the GPU that is used for scratch and small tensors
260
+ float tensor_split[128] = {0}; // how split tensors should be distributed across GPUs
261
+
262
+ enum llama_split_mode split_mode = LLAMA_SPLIT_MODE_LAYER; // how to split the model across GPUs
263
+
264
+ struct cpu_params cpuparams;
265
+ struct cpu_params cpuparams_batch;
266
+
267
+ lm_ggml_backend_sched_eval_callback cb_eval = nullptr;
268
+ void * cb_eval_user_data = nullptr;
269
+
270
+ lm_ggml_numa_strategy numa = LM_GGML_NUMA_STRATEGY_DISABLED;
271
+
272
+ enum llama_rope_scaling_type rope_scaling_type = LLAMA_ROPE_SCALING_TYPE_UNSPECIFIED;
273
+ enum llama_pooling_type pooling_type = LLAMA_POOLING_TYPE_UNSPECIFIED; // pooling type for embeddings
274
+ enum llama_attention_type attention_type = LLAMA_ATTENTION_TYPE_UNSPECIFIED; // attention type for embeddings
275
+
276
+ struct common_params_sampling sampling;
277
+ struct common_params_speculative speculative;
278
+ struct common_params_vocoder vocoder;
279
+
280
+ struct common_params_model model;
281
+
282
+ std::string model_alias = ""; // model alias // NOLINT
283
+ std::string hf_token = ""; // HF token // NOLINT
284
+ std::string prompt = ""; // NOLINT
285
+ std::string system_prompt = ""; // NOLINT
286
+ std::string prompt_file = ""; // store the external prompt file name // NOLINT
287
+ std::string path_prompt_cache = ""; // path to file for saving/loading prompt eval state // NOLINT
288
+ std::string input_prefix = ""; // string to prefix user inputs with // NOLINT
289
+ std::string input_suffix = ""; // string to suffix user inputs with // NOLINT
290
+ std::string lookup_cache_static = ""; // path of static ngram cache file for lookup decoding // NOLINT
291
+ std::string lookup_cache_dynamic = ""; // path of dynamic ngram cache file for lookup decoding // NOLINT
292
+ std::string logits_file = ""; // file for saving *all* logits // NOLINT
293
+
294
+ std::vector<std::string> in_files; // all input files
295
+ std::vector<std::string> antiprompt; // strings upon which more user input is prompted (a.k.a. reverse prompts)
296
+ std::vector<llama_model_kv_override> kv_overrides;
297
+ std::vector<llama_model_tensor_buft_override> tensor_buft_overrides;
298
+
299
+ bool lora_init_without_apply = false; // only load lora to memory, but do not apply it to ctx (user can manually apply lora later using llama_adapter_lora_apply)
300
+ std::vector<common_adapter_lora_info> lora_adapters; // lora adapter path with user defined scale
301
+
302
+ std::vector<common_control_vector_load_info> control_vectors; // control vector with user defined scale
303
+
304
+ int32_t verbosity = 0;
305
+ int32_t control_vector_layer_start = -1; // layer range for control vector
306
+ int32_t control_vector_layer_end = -1; // layer range for control vector
307
+
308
+ int32_t ppl_stride = 0; // stride for perplexity calculations. If left at 0, the pre-existing approach will be used.
309
+ int32_t ppl_output_type = 0; // = 0 -> ppl output is as usual, = 1 -> ppl output is num_tokens, ppl, one per line
310
+ // (which is more convenient to use for plotting)
311
+ //
312
+ bool hellaswag = false; // compute HellaSwag score over random tasks from datafile supplied in prompt
313
+ size_t hellaswag_tasks = 400; // number of tasks to use when computing the HellaSwag score
314
+
315
+ bool winogrande = false; // compute Winogrande score over random tasks from datafile supplied in prompt
316
+ size_t winogrande_tasks = 0; // number of tasks to use when computing the Winogrande score. If 0, all tasks will be computed
317
+
318
+ bool multiple_choice = false; // compute TruthfulQA score over random tasks from datafile supplied in prompt
319
+ size_t multiple_choice_tasks = 0; // number of tasks to use when computing the TruthfulQA score. If 0, all tasks will be computed
320
+
321
+ bool kl_divergence = false; // compute KL divergence
322
+
323
+ bool usage = false; // print usage
324
+ bool completion = false; // print source-able completion script
325
+ bool use_color = false; // use color to distinguish generations and inputs
326
+ bool special = false; // enable special token output
327
+ bool interactive = false; // interactive mode
328
+ bool interactive_first = false; // wait for user input immediately
329
+ bool prompt_cache_all = false; // save user input and generations to prompt cache
330
+ bool prompt_cache_ro = false; // open the prompt cache read-only and do not update it
331
+
332
+ bool escape = true; // escape "\n", "\r", "\t", "\'", "\"", and "\\"
333
+ bool multiline_input = false; // reverse the usage of `\`
334
+ bool simple_io = false; // improves compatibility with subprocesses and limited consoles
335
+ bool cont_batching = true; // insert new sequences for decoding on-the-fly
336
+ bool flash_attn = false; // flash attention
337
+ bool no_perf = false; // disable performance metrics
338
+ bool ctx_shift = true; // context shift on inifinite text generation
339
+
340
+ bool input_prefix_bos = false; // prefix BOS to user inputs, preceding input_prefix
341
+ bool logits_all = false; // return logits for all tokens in the batch
342
+ bool use_mmap = true; // use mmap for faster loads
343
+ bool use_mlock = false; // use mlock to keep model in memory
344
+ bool verbose_prompt = false; // print prompt tokens before generation
345
+ bool display_prompt = true; // print prompt before generation
346
+ bool dump_kv_cache = false; // dump the KV cache contents for debugging purposes
347
+ bool no_kv_offload = false; // disable KV offloading
348
+ bool warmup = true; // warmup run
349
+ bool check_tensors = false; // validate tensor data
350
+
351
+ bool single_turn = false; // single turn chat conversation
352
+
353
+ lm_ggml_type cache_type_k = LM_GGML_TYPE_F16; // KV cache data type for the K
354
+ lm_ggml_type cache_type_v = LM_GGML_TYPE_F16; // KV cache data type for the V
355
+
356
+ common_conversation_mode conversation_mode = COMMON_CONVERSATION_MODE_AUTO;
357
+
358
+ // multimodal models (see examples/llava)
359
+ struct common_params_model mmproj;
360
+ std::vector<std::string> image; // path to image file(s)
361
+
362
+ // embedding
363
+ bool embedding = false; // get only sentence embedding
364
+ int32_t embd_normalize = 2; // normalisation for embeddings (-1=none, 0=max absolute int16, 1=taxicab, 2=euclidean, >2=p-norm)
365
+ std::string embd_out = ""; // empty = default, "array" = [[],[]...], "json" = openai style, "json+" = same "json" + cosine similarity matrix
366
+ std::string embd_sep = "\n"; // separator of embeddings
367
+ bool reranking = false; // enable reranking support on server
368
+
369
+ // server params
370
+ int32_t port = 8080; // server listens on this network port
371
+ int32_t timeout_read = 600; // http read timeout in seconds
372
+ int32_t timeout_write = timeout_read; // http write timeout in seconds
373
+ int32_t n_threads_http = -1; // number of threads to process HTTP requests (TODO: support threadpool)
374
+ int32_t n_cache_reuse = 0; // min chunk size to reuse from the cache via KV shifting
375
+
376
+ std::string hostname = "127.0.0.1";
377
+ std::string public_path = ""; // NOLINT
378
+ std::string chat_template = ""; // NOLINT
379
+ bool use_jinja = false; // NOLINT
380
+ bool enable_chat_template = true;
381
+ common_reasoning_format reasoning_format = COMMON_REASONING_FORMAT_DEEPSEEK;
382
+
383
+ std::vector<std::string> api_keys;
384
+
385
+ std::string ssl_file_key = ""; // NOLINT
386
+ std::string ssl_file_cert = ""; // NOLINT
387
+
388
+ // "advanced" endpoints are disabled by default for better security
389
+ bool webui = true;
390
+ bool endpoint_slots = false;
391
+ bool endpoint_props = false; // only control POST requests, not GET
392
+ bool endpoint_metrics = false;
393
+
394
+ bool log_json = false;
395
+
396
+ std::string slot_save_path;
397
+
398
+ float slot_prompt_similarity = 0.5f;
399
+
400
+ // batched-bench params
401
+ bool is_pp_shared = false;
402
+
403
+ std::vector<int32_t> n_pp;
404
+ std::vector<int32_t> n_tg;
405
+ std::vector<int32_t> n_pl;
406
+
407
+ // retrieval params
408
+ std::vector<std::string> context_files; // context files to embed
409
+
410
+ int32_t chunk_size = 64; // chunk size for context embedding
411
+
412
+ std::string chunk_separator = "\n"; // chunk separator for context embedding
413
+
414
+ // passkey params
415
+ int32_t n_junk = 250; // number of times to repeat the junk text
416
+ int32_t i_pos = -1; // position of the passkey in the junk text
417
+
418
+ // imatrix params
419
+ int32_t n_out_freq = 10; // output the imatrix every n_out_freq iterations
420
+ int32_t n_save_freq = 0; // save the imatrix every n_save_freq iterations
421
+ int32_t i_chunk = 0; // start processing from this chunk
422
+
423
+ bool process_output = false; // collect data for the output tensor
424
+ bool compute_ppl = true; // whether to compute perplexity
425
+
426
+ // cvector-generator params
427
+ int n_pca_batch = 100;
428
+ int n_pca_iterations = 1000;
429
+ dimre_method cvector_dimre_method = DIMRE_METHOD_PCA;
430
+ std::string cvector_positive_file = "examples/cvector-generator/positive.txt";
431
+ std::string cvector_negative_file = "examples/cvector-generator/negative.txt";
432
+
433
+ bool spm_infill = false; // suffix/prefix/middle pattern for infill
434
+
435
+ // batched-bench params
436
+ bool batched_bench_output_jsonl = false;
437
+
438
+ // common params
439
+ std::string out_file; // output filename for all example programs
440
+ };
441
+
442
+ // call once at the start of a program if it uses libcommon
443
+ // initializes the logging system and prints info about the build
444
+ void common_init();
445
+
446
+ std::string common_params_get_system_info(const common_params & params);
447
+
448
+ bool parse_cpu_range(const std::string & range, bool(&boolmask)[LM_GGML_MAX_N_THREADS]);
449
+ bool parse_cpu_mask(const std::string & mask, bool(&boolmask)[LM_GGML_MAX_N_THREADS]);
450
+ void postprocess_cpu_params(cpu_params & cpuparams, const cpu_params * role_model = nullptr);
451
+ bool set_process_priority(enum lm_ggml_sched_priority prio);
452
+
453
+ //
454
+ // String utils
455
+ //
456
+
457
+ #ifdef __GNUC__
458
+ # if defined(__MINGW32__) && !defined(__clang__)
459
+ # define LLAMA_COMMON_ATTRIBUTE_FORMAT(...) __attribute__((format(gnu_printf, __VA_ARGS__)))
460
+ # else
461
+ # define LLAMA_COMMON_ATTRIBUTE_FORMAT(...) __attribute__((format(printf, __VA_ARGS__)))
462
+ # endif
463
+ #else
464
+ # define LLAMA_COMMON_ATTRIBUTE_FORMAT(...)
465
+ #endif
466
+
467
+ LLAMA_COMMON_ATTRIBUTE_FORMAT(1, 2)
468
+ std::string string_format(const char * fmt, ...);
469
+
470
+ std::string string_strip(const std::string & str);
471
+ std::string string_get_sortable_timestamp();
472
+
473
+ std::string string_join(const std::vector<std::string> & values, const std::string & separator);
474
+ std::vector<std::string> string_split(const std::string & str, const std::string & delimiter);
475
+ std::string string_repeat(const std::string & str, size_t n);
476
+
477
+ void string_replace_all(std::string & s, const std::string & search, const std::string & replace);
478
+
479
+ std::string regex_escape(const std::string & s);
480
+
481
+ template<class T>
482
+ static std::vector<T> string_split(const std::string & str, char delim) {
483
+ static_assert(!std::is_same<T, std::string>::value, "Please use the specialized version for std::string");
484
+ std::vector<T> values;
485
+ std::istringstream str_stream(str);
486
+ std::string token;
487
+ while (std::getline(str_stream, token, delim)) {
488
+ T value;
489
+ std::istringstream token_stream(token);
490
+ token_stream >> value;
491
+ values.push_back(value);
492
+ }
493
+ return values;
494
+ }
495
+
496
+ template<>
497
+ std::vector<std::string> string_split<std::string>(const std::string & input, char separator)
498
+ {
499
+ std::vector<std::string> parts;
500
+ size_t begin_pos = 0;
501
+ size_t separator_pos = input.find(separator);
502
+ while (separator_pos != std::string::npos) {
503
+ std::string part = input.substr(begin_pos, separator_pos - begin_pos);
504
+ parts.emplace_back(part);
505
+ begin_pos = separator_pos + 1;
506
+ separator_pos = input.find(separator, begin_pos);
507
+ }
508
+ parts.emplace_back(input.substr(begin_pos, separator_pos - begin_pos));
509
+ return parts;
510
+ }
511
+
512
+ static bool string_starts_with(const std::string & str,
513
+ const std::string & prefix) { // While we wait for C++20's std::string::starts_with...
514
+ return str.rfind(prefix, 0) == 0;
515
+ }
516
+
517
+ static bool string_ends_with(const std::string & str,
518
+ const std::string & suffix) { // While we wait for C++20's std::string::ends_with...
519
+ return str.size() >= suffix.size() && str.compare(str.size()-suffix.size(), suffix.size(), suffix) == 0;
520
+ }
521
+
522
+ bool string_parse_kv_override(const char * data, std::vector<llama_model_kv_override> & overrides);
523
+ void string_process_escapes(std::string & input);
524
+
525
+ std::string string_from(bool value);
526
+ std::string string_from(const std::vector<int> & values);
527
+ std::string string_from(const struct llama_context * ctx, const std::vector<llama_token> & tokens);
528
+ std::string string_from(const struct llama_context * ctx, const struct llama_batch & batch);
529
+
530
+ //
531
+ // Filesystem utils
532
+ //
533
+
534
+ bool fs_validate_filename(const std::string & filename);
535
+ bool fs_create_directory_with_parents(const std::string & path);
536
+
537
+ std::string fs_get_cache_directory();
538
+ std::string fs_get_cache_file(const std::string & filename);
539
+
540
+ //
541
+ // Model utils
542
+ //
543
+
544
+ // note: defines object's lifetime
545
+ struct common_init_result {
546
+ llama_model_ptr model;
547
+ llama_context_ptr context;
548
+
549
+ std::vector<llama_adapter_lora_ptr> lora;
550
+ };
551
+
552
+ struct common_init_result common_init_from_params(common_params & params);
553
+
554
+ struct llama_model_params common_model_params_to_llama ( common_params & params);
555
+ struct llama_context_params common_context_params_to_llama(const common_params & params);
556
+ struct lm_ggml_threadpool_params lm_ggml_threadpool_params_from_cpu_params(const cpu_params & params);
557
+
558
+ // clear LoRA adapters from context, then apply new list of adapters
559
+ void common_set_adapter_lora(struct llama_context * ctx, std::vector<common_adapter_lora_info> & lora);
560
+
561
+ //
562
+ // Batch utils
563
+ //
564
+
565
+ void common_batch_clear(struct llama_batch & batch);
566
+
567
+ void common_batch_add(
568
+ struct llama_batch & batch,
569
+ llama_token id,
570
+ llama_pos pos,
571
+ const std::vector<llama_seq_id> & seq_ids,
572
+ bool logits);
573
+
574
+ //
575
+ // Token utils
576
+ //
577
+
578
+ // longest common prefix
579
+ size_t common_lcp(const llama_tokens & a, const llama_tokens & b);
580
+
581
+ // longet common subsequence
582
+ size_t common_lcs(const llama_tokens & a, const llama_tokens & b);
583
+
584
+ //
585
+ // Vocab utils
586
+ //
587
+
588
+ // tokenizes a string into a vector of tokens
589
+ // should work similar to Python's `tokenizer.encode`
590
+ std::vector<llama_token> common_tokenize(
591
+ const struct llama_context * ctx,
592
+ const std::string & text,
593
+ bool add_special,
594
+ bool parse_special = false);
595
+
596
+ std::vector<llama_token> common_tokenize(
597
+ const struct llama_vocab * vocab,
598
+ const std::string & text,
599
+ bool add_special,
600
+ bool parse_special = false);
601
+
602
+ // tokenizes a token into a piece, optionally renders special/control tokens
603
+ // should work similar to Python's `tokenizer.id_to_piece`
604
+ std::string common_token_to_piece(
605
+ const struct llama_context * ctx,
606
+ llama_token token,
607
+ bool special = true);
608
+
609
+ std::string common_token_to_piece(
610
+ const struct llama_vocab * vocab,
611
+ llama_token token,
612
+ bool special = true);
613
+
614
+ // detokenizes a vector of tokens into a string
615
+ // should work similar to Python's `tokenizer.decode`
616
+ // optionally renders special/control tokens
617
+ std::string common_detokenize(
618
+ const struct llama_context * ctx,
619
+ const std::vector<llama_token> & tokens,
620
+ bool special = true);
621
+
622
+ std::string common_detokenize(
623
+ const struct llama_vocab * vocab,
624
+ const std::vector<llama_token> & tokens,
625
+ bool special = true);
626
+
627
+ //
628
+ // KV cache utils
629
+ //
630
+
631
+ // Dump the KV cache view with the number of sequences per cell.
632
+ void common_kv_cache_dump_view(const llama_kv_cache_view & view, int row_size = 80);
633
+
634
+ // Dump the KV cache view showing individual sequences in each cell (long output).
635
+ void common_kv_cache_dump_view_seqs(const llama_kv_cache_view & view, int row_size = 40);
636
+
637
+ //
638
+ // Embedding utils
639
+ //
640
+
641
+ // TODO: repace embd_norm with an enum
642
+ void common_embd_normalize(const float * inp, float * out, int n, int embd_norm);
643
+
644
+ float common_embd_similarity_cos(const float * embd1, const float * embd2, int n);
645
+
646
+ //
647
+ // Control vector utils
648
+ //
649
+
650
+ struct common_control_vector_data {
651
+ int n_embd;
652
+
653
+ // stores data for layers [1, n_layer] where n_layer = data.size() / n_embd
654
+ std::vector<float> data;
655
+ };
656
+
657
+ struct common_control_vector_load_info {
658
+ float strength;
659
+
660
+ std::string fname;
661
+ };
662
+
663
+ // Load control vectors, scale each by strength, and add them together.
664
+ // On error, returns {-1, empty}
665
+ common_control_vector_data common_control_vector_load(const std::vector<common_control_vector_load_info> & load_infos);
666
+
667
+ //
668
+ // Split utils
669
+ //
670
+
671
+ namespace {
672
+
673
+ const char * const LLM_KV_SPLIT_NO = "split.no";
674
+ const char * const LLM_KV_SPLIT_COUNT = "split.count";
675
+ const char * const LLM_KV_SPLIT_TENSORS_COUNT = "split.tensors.count";
676
+
677
+ }