react-native-executorch 0.5.6 → 0.5.8

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 (237) hide show
  1. package/android/libs/classes.jar +0 -0
  2. package/android/src/main/cpp/CMakeLists.txt +23 -14
  3. package/common/rnexecutorch/RnExecutorchInstaller.cpp +4 -21
  4. package/common/rnexecutorch/host_objects/ModelHostObject.h +67 -51
  5. package/common/rnexecutorch/models/llm/LLM.cpp +24 -1
  6. package/common/rnexecutorch/models/llm/LLM.h +4 -1
  7. package/common/rnexecutorch/threads/GlobalThreadPool.h +79 -0
  8. package/common/rnexecutorch/threads/HighPerformanceThreadPool.h +364 -0
  9. package/common/rnexecutorch/threads/utils/ThreadUtils.h +29 -0
  10. package/common/runner/runner.cpp +9 -3
  11. package/common/runner/runner.h +4 -3
  12. package/common/runner/text_token_generator.h +28 -10
  13. package/lib/Error.js +53 -0
  14. package/lib/ThreadPool.d.ts +10 -0
  15. package/lib/ThreadPool.js +28 -0
  16. package/lib/common/Logger.d.ts +8 -0
  17. package/lib/common/Logger.js +19 -0
  18. package/lib/constants/directories.js +2 -0
  19. package/lib/constants/llmDefaults.d.ts +6 -0
  20. package/lib/constants/llmDefaults.js +16 -0
  21. package/lib/constants/modelUrls.d.ts +223 -0
  22. package/lib/constants/modelUrls.js +322 -0
  23. package/lib/constants/ocr/models.d.ts +882 -0
  24. package/lib/constants/ocr/models.js +182 -0
  25. package/lib/constants/ocr/symbols.js +139 -0
  26. package/lib/constants/sttDefaults.d.ts +28 -0
  27. package/lib/constants/sttDefaults.js +68 -0
  28. package/lib/controllers/LLMController.d.ts +47 -0
  29. package/lib/controllers/LLMController.js +213 -0
  30. package/lib/controllers/OCRController.js +67 -0
  31. package/lib/controllers/SpeechToTextController.d.ts +56 -0
  32. package/lib/controllers/SpeechToTextController.js +349 -0
  33. package/lib/controllers/VerticalOCRController.js +70 -0
  34. package/lib/hooks/computer_vision/useClassification.d.ts +15 -0
  35. package/lib/hooks/computer_vision/useClassification.js +7 -0
  36. package/lib/hooks/computer_vision/useImageEmbeddings.d.ts +15 -0
  37. package/lib/hooks/computer_vision/useImageEmbeddings.js +7 -0
  38. package/lib/hooks/computer_vision/useImageSegmentation.d.ts +38 -0
  39. package/lib/hooks/computer_vision/useImageSegmentation.js +7 -0
  40. package/lib/hooks/computer_vision/useOCR.d.ts +20 -0
  41. package/lib/hooks/computer_vision/useOCR.js +41 -0
  42. package/lib/hooks/computer_vision/useObjectDetection.d.ts +15 -0
  43. package/lib/hooks/computer_vision/useObjectDetection.js +7 -0
  44. package/lib/hooks/computer_vision/useStyleTransfer.d.ts +15 -0
  45. package/lib/hooks/computer_vision/useStyleTransfer.js +7 -0
  46. package/lib/hooks/computer_vision/useVerticalOCR.d.ts +21 -0
  47. package/lib/hooks/computer_vision/useVerticalOCR.js +43 -0
  48. package/lib/hooks/general/useExecutorchModule.d.ts +13 -0
  49. package/lib/hooks/general/useExecutorchModule.js +7 -0
  50. package/lib/hooks/natural_language_processing/useLLM.d.ts +10 -0
  51. package/lib/hooks/natural_language_processing/useLLM.js +78 -0
  52. package/lib/hooks/natural_language_processing/useSpeechToText.d.ts +27 -0
  53. package/lib/hooks/natural_language_processing/useSpeechToText.js +49 -0
  54. package/lib/hooks/natural_language_processing/useTextEmbeddings.d.ts +16 -0
  55. package/lib/hooks/natural_language_processing/useTextEmbeddings.js +7 -0
  56. package/lib/hooks/natural_language_processing/useTokenizer.d.ts +17 -0
  57. package/lib/hooks/natural_language_processing/useTokenizer.js +52 -0
  58. package/lib/hooks/useModule.js +45 -0
  59. package/lib/hooks/useNonStaticModule.d.ts +20 -0
  60. package/lib/hooks/useNonStaticModule.js +49 -0
  61. package/lib/index.d.ts +48 -0
  62. package/lib/index.js +58 -0
  63. package/lib/module/controllers/LLMController.js +21 -2
  64. package/lib/module/controllers/LLMController.js.map +1 -1
  65. package/lib/module/hooks/natural_language_processing/useLLM.js +6 -2
  66. package/lib/module/hooks/natural_language_processing/useLLM.js.map +1 -1
  67. package/lib/module/modules/natural_language_processing/LLMModule.js +7 -2
  68. package/lib/module/modules/natural_language_processing/LLMModule.js.map +1 -1
  69. package/lib/module/types/llm.js.map +1 -1
  70. package/lib/modules/BaseModule.js +25 -0
  71. package/lib/modules/BaseNonStaticModule.js +14 -0
  72. package/lib/modules/computer_vision/ClassificationModule.d.ts +8 -0
  73. package/lib/modules/computer_vision/ClassificationModule.js +17 -0
  74. package/lib/modules/computer_vision/ImageEmbeddingsModule.d.ts +8 -0
  75. package/lib/modules/computer_vision/ImageEmbeddingsModule.js +17 -0
  76. package/lib/modules/computer_vision/ImageSegmentationModule.d.ts +11 -0
  77. package/lib/modules/computer_vision/ImageSegmentationModule.js +27 -0
  78. package/lib/modules/computer_vision/OCRModule.d.ts +14 -0
  79. package/lib/modules/computer_vision/OCRModule.js +17 -0
  80. package/lib/modules/computer_vision/ObjectDetectionModule.d.ts +9 -0
  81. package/lib/modules/computer_vision/ObjectDetectionModule.js +17 -0
  82. package/lib/modules/computer_vision/StyleTransferModule.d.ts +8 -0
  83. package/lib/modules/computer_vision/StyleTransferModule.js +17 -0
  84. package/lib/modules/computer_vision/VerticalOCRModule.d.ts +14 -0
  85. package/lib/modules/computer_vision/VerticalOCRModule.js +19 -0
  86. package/lib/modules/general/ExecutorchModule.d.ts +7 -0
  87. package/lib/modules/general/ExecutorchModule.js +14 -0
  88. package/lib/modules/natural_language_processing/LLMModule.d.ts +28 -0
  89. package/lib/modules/natural_language_processing/LLMModule.js +45 -0
  90. package/lib/modules/natural_language_processing/SpeechToTextModule.d.ts +24 -0
  91. package/lib/modules/natural_language_processing/SpeechToTextModule.js +36 -0
  92. package/lib/modules/natural_language_processing/TextEmbeddingsModule.d.ts +9 -0
  93. package/lib/modules/natural_language_processing/TextEmbeddingsModule.js +21 -0
  94. package/lib/modules/natural_language_processing/TokenizerModule.d.ts +12 -0
  95. package/lib/modules/natural_language_processing/TokenizerModule.js +30 -0
  96. package/lib/native/NativeETInstaller.js +2 -0
  97. package/lib/native/NativeOCR.js +2 -0
  98. package/lib/native/NativeVerticalOCR.js +2 -0
  99. package/lib/native/RnExecutorchModules.d.ts +7 -0
  100. package/lib/native/RnExecutorchModules.js +18 -0
  101. package/lib/tsconfig.tsbuildinfo +1 -0
  102. package/lib/types/common.d.ts +32 -0
  103. package/lib/types/common.js +25 -0
  104. package/lib/types/imageSegmentation.js +26 -0
  105. package/lib/types/llm.d.ts +46 -0
  106. package/lib/types/llm.js +9 -0
  107. package/lib/types/objectDetection.js +94 -0
  108. package/lib/types/ocr.js +1 -0
  109. package/lib/types/stt.d.ts +94 -0
  110. package/lib/types/stt.js +85 -0
  111. package/lib/typescript/controllers/LLMController.d.ts +4 -2
  112. package/lib/typescript/controllers/LLMController.d.ts.map +1 -1
  113. package/lib/typescript/hooks/natural_language_processing/useLLM.d.ts.map +1 -1
  114. package/lib/typescript/modules/natural_language_processing/LLMModule.d.ts +4 -2
  115. package/lib/typescript/modules/natural_language_processing/LLMModule.d.ts.map +1 -1
  116. package/lib/typescript/types/llm.d.ts +7 -1
  117. package/lib/typescript/types/llm.d.ts.map +1 -1
  118. package/lib/utils/ResourceFetcher.d.ts +24 -0
  119. package/lib/utils/ResourceFetcher.js +305 -0
  120. package/lib/utils/ResourceFetcherUtils.d.ts +54 -0
  121. package/lib/utils/ResourceFetcherUtils.js +127 -0
  122. package/lib/utils/llm.d.ts +6 -0
  123. package/lib/utils/llm.js +72 -0
  124. package/lib/utils/stt.js +21 -0
  125. package/package.json +3 -1
  126. package/react-native-executorch.podspec +12 -31
  127. package/src/controllers/LLMController.ts +29 -5
  128. package/src/hooks/natural_language_processing/useLLM.ts +15 -1
  129. package/src/modules/natural_language_processing/LLMModule.ts +14 -2
  130. package/src/types/llm.ts +8 -0
  131. package/third-party/android/libs/cpuinfo/arm64-v8a/libcpuinfo.so +0 -0
  132. package/third-party/android/libs/executorch/arm64-v8a/libexecutorch.so +0 -0
  133. package/third-party/android/libs/executorch/x86_64/libexecutorch.so +0 -0
  134. package/third-party/android/libs/pthreadpool/arm64-v8a/libpthreadpool.so +0 -0
  135. package/third-party/android/libs/tokenizers-cpp/arm64-v8a/libsentencepiece.a +0 -0
  136. package/third-party/android/libs/tokenizers-cpp/arm64-v8a/libtokenizers_c.a +0 -0
  137. package/third-party/android/libs/tokenizers-cpp/arm64-v8a/libtokenizers_cpp.a +0 -0
  138. package/third-party/android/libs/tokenizers-cpp/x86_64/libsentencepiece.a +0 -0
  139. package/third-party/android/libs/tokenizers-cpp/x86_64/libtokenizers_c.a +0 -0
  140. package/third-party/android/libs/tokenizers-cpp/x86_64/libtokenizers_cpp.a +0 -0
  141. package/third-party/include/c10/macros/Export.h +2 -86
  142. package/third-party/include/c10/macros/Macros.h +28 -5
  143. package/third-party/include/c10/util/BFloat16-inl.h +1 -4
  144. package/third-party/include/c10/util/BFloat16.h +5 -8
  145. package/third-party/include/c10/util/Half.h +5 -0
  146. package/third-party/include/c10/util/bit_cast.h +1 -1
  147. package/third-party/include/c10/util/complex.h +639 -0
  148. package/third-party/include/c10/util/complex_math.h +399 -0
  149. package/third-party/include/c10/util/complex_utils.h +41 -0
  150. package/third-party/include/c10/util/irange.h +2 -2
  151. package/third-party/include/c10/util/overflows.h +95 -0
  152. package/third-party/include/executorch/ExecuTorchError.h +75 -0
  153. package/third-party/include/executorch/ExecuTorchModule.h +115 -11
  154. package/third-party/include/executorch/ExecuTorchTensor.h +731 -51
  155. package/third-party/include/executorch/ExecuTorchValue.h +61 -9
  156. package/third-party/include/executorch/extension/kernel_util/make_boxed_from_unboxed_functor.h +181 -0
  157. package/third-party/include/executorch/extension/kernel_util/meta_programming.h +108 -0
  158. package/third-party/include/executorch/extension/kernel_util/type_list.h +137 -0
  159. package/third-party/include/executorch/extension/module/bundled_module.h +131 -0
  160. package/third-party/include/executorch/extension/module/module.h +46 -20
  161. package/third-party/include/executorch/extension/threadpool/cpuinfo_utils.h +1 -3
  162. package/third-party/include/executorch/extension/threadpool/threadpool.h +1 -3
  163. package/third-party/include/executorch/extension/threadpool/threadpool_guard.h +35 -0
  164. package/third-party/include/executorch/runtime/backend/backend_execution_context.h +3 -3
  165. package/third-party/include/executorch/runtime/backend/backend_init_context.h +12 -6
  166. package/third-party/include/executorch/runtime/backend/backend_option_context.h +34 -0
  167. package/third-party/include/executorch/runtime/backend/interface.h +70 -9
  168. package/third-party/include/executorch/runtime/backend/options.h +206 -0
  169. package/third-party/include/executorch/runtime/core/evalue.h +19 -25
  170. package/third-party/include/executorch/runtime/core/event_tracer.h +32 -17
  171. package/third-party/include/executorch/runtime/core/event_tracer_hooks.h +23 -14
  172. package/third-party/include/executorch/runtime/core/exec_aten/exec_aten.h +32 -9
  173. package/third-party/include/executorch/runtime/core/exec_aten/util/dim_order_util.h +3 -2
  174. package/third-party/include/executorch/runtime/core/exec_aten/util/scalar_type_util.h +43 -75
  175. package/third-party/include/executorch/runtime/core/exec_aten/util/tensor_util.h +88 -87
  176. package/third-party/include/executorch/runtime/core/function_ref.h +100 -0
  177. package/third-party/include/executorch/runtime/core/named_data_map.h +14 -14
  178. package/third-party/include/executorch/runtime/core/portable_type/c10/c10/macros/Export.h +2 -86
  179. package/third-party/include/executorch/runtime/core/portable_type/c10/c10/macros/Macros.h +28 -5
  180. package/third-party/include/executorch/runtime/core/portable_type/c10/c10/util/BFloat16-inl.h +1 -4
  181. package/third-party/include/executorch/runtime/core/portable_type/c10/c10/util/BFloat16.h +5 -8
  182. package/third-party/include/executorch/runtime/core/portable_type/c10/c10/util/Half.h +5 -0
  183. package/third-party/include/executorch/runtime/core/portable_type/c10/c10/util/bit_cast.h +1 -1
  184. package/third-party/include/executorch/runtime/core/portable_type/c10/c10/util/complex.h +639 -0
  185. package/third-party/include/executorch/runtime/core/portable_type/c10/c10/util/complex_math.h +399 -0
  186. package/third-party/include/executorch/runtime/core/portable_type/c10/c10/util/complex_utils.h +41 -0
  187. package/third-party/include/executorch/runtime/core/portable_type/c10/c10/util/irange.h +2 -2
  188. package/third-party/include/executorch/runtime/core/portable_type/c10/c10/util/overflows.h +95 -0
  189. package/third-party/include/executorch/runtime/core/portable_type/c10/torch/headeronly/macros/Export.h +88 -0
  190. package/third-party/include/executorch/runtime/core/portable_type/complex.h +6 -29
  191. package/third-party/include/executorch/runtime/core/portable_type/tensor_impl.h +20 -0
  192. package/third-party/include/executorch/runtime/core/span.h +4 -0
  193. package/third-party/include/executorch/runtime/core/tag.h +19 -0
  194. package/third-party/include/executorch/runtime/core/tensor_layout.h +2 -2
  195. package/third-party/include/executorch/runtime/executor/method.h +15 -3
  196. package/third-party/include/executorch/runtime/executor/method_meta.h +34 -5
  197. package/third-party/include/executorch/runtime/executor/program.h +3 -4
  198. package/third-party/include/executorch/runtime/executor/pte_data_map.h +9 -8
  199. package/third-party/include/executorch/runtime/executor/tensor_parser.h +14 -13
  200. package/third-party/include/executorch/runtime/kernel/kernel_runtime_context.h +5 -5
  201. package/third-party/include/executorch/runtime/kernel/operator_registry.h +21 -19
  202. package/third-party/include/executorch/runtime/platform/compiler.h +8 -0
  203. package/third-party/include/executorch/runtime/platform/platform.h +126 -0
  204. package/third-party/include/headeronly/macros/Export.h +88 -0
  205. package/third-party/include/tokenizers-cpp/tokenizers_c.h +61 -0
  206. package/third-party/include/torch/headeronly/macros/Export.h +88 -0
  207. package/third-party/ios/ExecutorchLib.xcframework/Info.plist +43 -0
  208. package/third-party/ios/ExecutorchLib.xcframework/ios-arm64/ExecutorchLib.framework/ExecutorchLib +0 -0
  209. package/third-party/ios/ExecutorchLib.xcframework/ios-arm64/ExecutorchLib.framework/Info.plist +0 -0
  210. package/third-party/ios/ExecutorchLib.xcframework/ios-arm64-simulator/ExecutorchLib.framework/ExecutorchLib +0 -0
  211. package/third-party/ios/ExecutorchLib.xcframework/ios-arm64-simulator/ExecutorchLib.framework/Info.plist +0 -0
  212. package/third-party/ios/libs/cpuinfo/libcpuinfo.a +0 -0
  213. package/third-party/ios/libs/pthreadpool/physical-arm64-release/libpthreadpool.a +0 -0
  214. package/third-party/ios/libs/pthreadpool/simulator-arm64-debug/libpthreadpool.a +0 -0
  215. package/ios/libs/executorch/libbackend_coreml_ios.a +0 -0
  216. package/ios/libs/executorch/libbackend_coreml_simulator.a +0 -0
  217. package/ios/libs/executorch/libbackend_mps_ios.a +0 -0
  218. package/ios/libs/executorch/libbackend_mps_simulator.a +0 -0
  219. package/ios/libs/executorch/libbackend_xnnpack_ios.a +0 -0
  220. package/ios/libs/executorch/libbackend_xnnpack_simulator.a +0 -0
  221. package/ios/libs/executorch/libexecutorch_ios.a +0 -0
  222. package/ios/libs/executorch/libexecutorch_simulator.a +0 -0
  223. package/ios/libs/executorch/libkernels_custom_ios.a +0 -0
  224. package/ios/libs/executorch/libkernels_custom_simulator.a +0 -0
  225. package/ios/libs/executorch/libkernels_optimized_ios.a +0 -0
  226. package/ios/libs/executorch/libkernels_optimized_simulator.a +0 -0
  227. package/ios/libs/executorch/libkernels_portable_ios.a +0 -0
  228. package/ios/libs/executorch/libkernels_portable_simulator.a +0 -0
  229. package/ios/libs/executorch/libkernels_quantized_ios.a +0 -0
  230. package/ios/libs/executorch/libkernels_quantized_simulator.a +0 -0
  231. package/third-party/ios/ios.toolchain.cmake +0 -1122
  232. /package/{ios → third-party/ios}/libs/tokenizers-cpp/physical-arm64-release/libsentencepiece.a +0 -0
  233. /package/{ios → third-party/ios}/libs/tokenizers-cpp/physical-arm64-release/libtokenizers_c.a +0 -0
  234. /package/{ios → third-party/ios}/libs/tokenizers-cpp/physical-arm64-release/libtokenizers_cpp.a +0 -0
  235. /package/{ios → third-party/ios}/libs/tokenizers-cpp/simulator-arm64-debug/libsentencepiece.a +0 -0
  236. /package/{ios → third-party/ios}/libs/tokenizers-cpp/simulator-arm64-debug/libtokenizers_c.a +0 -0
  237. /package/{ios → third-party/ios}/libs/tokenizers-cpp/simulator-arm64-debug/libtokenizers_cpp.a +0 -0
@@ -0,0 +1,223 @@
1
+ import { AvailableModels } from '../types/stt';
2
+ export declare const LLAMA3_2_3B: {
3
+ modelSource: string;
4
+ tokenizerSource: string;
5
+ tokenizerConfigSource: string;
6
+ };
7
+ export declare const LLAMA3_2_3B_QLORA: {
8
+ modelSource: string;
9
+ tokenizerSource: string;
10
+ tokenizerConfigSource: string;
11
+ };
12
+ export declare const LLAMA3_2_3B_SPINQUANT: {
13
+ modelSource: string;
14
+ tokenizerSource: string;
15
+ tokenizerConfigSource: string;
16
+ };
17
+ export declare const LLAMA3_2_1B: {
18
+ modelSource: string;
19
+ tokenizerSource: string;
20
+ tokenizerConfigSource: string;
21
+ };
22
+ export declare const LLAMA3_2_1B_QLORA: {
23
+ modelSource: string;
24
+ tokenizerSource: string;
25
+ tokenizerConfigSource: string;
26
+ };
27
+ export declare const LLAMA3_2_1B_SPINQUANT: {
28
+ modelSource: string;
29
+ tokenizerSource: string;
30
+ tokenizerConfigSource: string;
31
+ };
32
+ export declare const QWEN3_0_6B: {
33
+ modelSource: string;
34
+ tokenizerSource: string;
35
+ tokenizerConfigSource: string;
36
+ };
37
+ export declare const QWEN3_0_6B_QUANTIZED: {
38
+ modelSource: string;
39
+ tokenizerSource: string;
40
+ tokenizerConfigSource: string;
41
+ };
42
+ export declare const QWEN3_1_7B: {
43
+ modelSource: string;
44
+ tokenizerSource: string;
45
+ tokenizerConfigSource: string;
46
+ };
47
+ export declare const QWEN3_1_7B_QUANTIZED: {
48
+ modelSource: string;
49
+ tokenizerSource: string;
50
+ tokenizerConfigSource: string;
51
+ };
52
+ export declare const QWEN3_4B: {
53
+ modelSource: string;
54
+ tokenizerSource: string;
55
+ tokenizerConfigSource: string;
56
+ };
57
+ export declare const QWEN3_4B_QUANTIZED: {
58
+ modelSource: string;
59
+ tokenizerSource: string;
60
+ tokenizerConfigSource: string;
61
+ };
62
+ export declare const HAMMER2_1_0_5B: {
63
+ modelSource: string;
64
+ tokenizerSource: string;
65
+ tokenizerConfigSource: string;
66
+ };
67
+ export declare const HAMMER2_1_0_5B_QUANTIZED: {
68
+ modelSource: string;
69
+ tokenizerSource: string;
70
+ tokenizerConfigSource: string;
71
+ };
72
+ export declare const HAMMER2_1_1_5B: {
73
+ modelSource: string;
74
+ tokenizerSource: string;
75
+ tokenizerConfigSource: string;
76
+ };
77
+ export declare const HAMMER2_1_1_5B_QUANTIZED: {
78
+ modelSource: string;
79
+ tokenizerSource: string;
80
+ tokenizerConfigSource: string;
81
+ };
82
+ export declare const HAMMER2_1_3B: {
83
+ modelSource: string;
84
+ tokenizerSource: string;
85
+ tokenizerConfigSource: string;
86
+ };
87
+ export declare const HAMMER2_1_3B_QUANTIZED: {
88
+ modelSource: string;
89
+ tokenizerSource: string;
90
+ tokenizerConfigSource: string;
91
+ };
92
+ export declare const SMOLLM2_1_135M: {
93
+ modelSource: string;
94
+ tokenizerSource: string;
95
+ tokenizerConfigSource: string;
96
+ };
97
+ export declare const SMOLLM2_1_135M_QUANTIZED: {
98
+ modelSource: string;
99
+ tokenizerSource: string;
100
+ tokenizerConfigSource: string;
101
+ };
102
+ export declare const SMOLLM2_1_360M: {
103
+ modelSource: string;
104
+ tokenizerSource: string;
105
+ tokenizerConfigSource: string;
106
+ };
107
+ export declare const SMOLLM2_1_360M_QUANTIZED: {
108
+ modelSource: string;
109
+ tokenizerSource: string;
110
+ tokenizerConfigSource: string;
111
+ };
112
+ export declare const SMOLLM2_1_1_7B: {
113
+ modelSource: string;
114
+ tokenizerSource: string;
115
+ tokenizerConfigSource: string;
116
+ };
117
+ export declare const SMOLLM2_1_1_7B_QUANTIZED: {
118
+ modelSource: string;
119
+ tokenizerSource: string;
120
+ tokenizerConfigSource: string;
121
+ };
122
+ export declare const QWEN2_5_0_5B: {
123
+ modelSource: string;
124
+ tokenizerSource: string;
125
+ tokenizerConfigSource: string;
126
+ };
127
+ export declare const QWEN2_5_0_5B_QUANTIZED: {
128
+ modelSource: string;
129
+ tokenizerSource: string;
130
+ tokenizerConfigSource: string;
131
+ };
132
+ export declare const QWEN2_5_1_5B: {
133
+ modelSource: string;
134
+ tokenizerSource: string;
135
+ tokenizerConfigSource: string;
136
+ };
137
+ export declare const QWEN2_5_1_5B_QUANTIZED: {
138
+ modelSource: string;
139
+ tokenizerSource: string;
140
+ tokenizerConfigSource: string;
141
+ };
142
+ export declare const QWEN2_5_3B: {
143
+ modelSource: string;
144
+ tokenizerSource: string;
145
+ tokenizerConfigSource: string;
146
+ };
147
+ export declare const QWEN2_5_3B_QUANTIZED: {
148
+ modelSource: string;
149
+ tokenizerSource: string;
150
+ tokenizerConfigSource: string;
151
+ };
152
+ export declare const PHI_4_MINI_4B: {
153
+ modelSource: string;
154
+ tokenizerSource: string;
155
+ tokenizerConfigSource: string;
156
+ };
157
+ export declare const PHI_4_MINI_4B_QUANTIZED: {
158
+ modelSource: string;
159
+ tokenizerSource: string;
160
+ tokenizerConfigSource: string;
161
+ };
162
+ export declare const EFFICIENTNET_V2_S: {
163
+ modelSource: string;
164
+ };
165
+ export declare const SSDLITE_320_MOBILENET_V3_LARGE: {
166
+ modelSource: string;
167
+ };
168
+ export declare const STYLE_TRANSFER_CANDY: {
169
+ modelSource: string;
170
+ };
171
+ export declare const STYLE_TRANSFER_MOSAIC: {
172
+ modelSource: string;
173
+ };
174
+ export declare const STYLE_TRANSFER_RAIN_PRINCESS: {
175
+ modelSource: string;
176
+ };
177
+ export declare const STYLE_TRANSFER_UDNIE: {
178
+ modelSource: string;
179
+ };
180
+ export declare const MOONSHINE_TINY: {
181
+ modelName: AvailableModels;
182
+ decoderSource: string;
183
+ encoderSource: string;
184
+ tokenizerSource: string;
185
+ };
186
+ export declare const WHISPER_TINY: {
187
+ modelName: AvailableModels;
188
+ decoderSource: string;
189
+ encoderSource: string;
190
+ tokenizerSource: string;
191
+ };
192
+ export declare const WHISPER_TINY_MULTILINGUAL: {
193
+ modelName: AvailableModels;
194
+ decoderSource: string;
195
+ encoderSource: string;
196
+ tokenizerSource: string;
197
+ };
198
+ export declare const DEEPLAB_V3_RESNET50: {
199
+ modelSource: string;
200
+ };
201
+ export declare const CLIP_VIT_BASE_PATCH32_IMAGE: {
202
+ modelSource: string;
203
+ };
204
+ export declare const ALL_MINILM_L6_V2: {
205
+ modelSource: string;
206
+ tokenizerSource: string;
207
+ };
208
+ export declare const ALL_MPNET_BASE_V2: {
209
+ modelSource: string;
210
+ tokenizerSource: string;
211
+ };
212
+ export declare const MULTI_QA_MINILM_L6_COS_V1: {
213
+ modelSource: string;
214
+ tokenizerSource: string;
215
+ };
216
+ export declare const MULTI_QA_MPNET_BASE_DOT_V1: {
217
+ modelSource: string;
218
+ tokenizerSource: string;
219
+ };
220
+ export declare const CLIP_VIT_BASE_PATCH32_TEXT: {
221
+ modelSource: string;
222
+ tokenizerSource: string;
223
+ };
@@ -0,0 +1,322 @@
1
+ import { Platform } from 'react-native';
2
+ import { AvailableModels } from '../types/stt';
3
+ const URL_PREFIX = 'https://huggingface.co/software-mansion/react-native-executorch';
4
+ const VERSION_TAG = 'resolve/v0.5.0';
5
+ const NEXT_VERSION_TAG = 'resolve/v0.5.0';
6
+ // LLMs
7
+ // LLAMA 3.2
8
+ const LLAMA3_2_3B_MODEL = `${URL_PREFIX}-llama-3.2/${VERSION_TAG}/llama-3.2-3B/original/llama3_2_3B_bf16.pte`;
9
+ const LLAMA3_2_3B_QLORA_MODEL = `${URL_PREFIX}-llama-3.2/${VERSION_TAG}/llama-3.2-3B/QLoRA/llama3_2-3B_qat_lora.pte`;
10
+ const LLAMA3_2_3B_SPINQUANT_MODEL = `${URL_PREFIX}-llama-3.2/${VERSION_TAG}/llama-3.2-3B/spinquant/llama3_2_3B_spinquant.pte`;
11
+ const LLAMA3_2_1B_MODEL = `${URL_PREFIX}-llama-3.2/${VERSION_TAG}/llama-3.2-1B/original/llama3_2_bf16.pte`;
12
+ const LLAMA3_2_1B_QLORA_MODEL = `${URL_PREFIX}-llama-3.2/${VERSION_TAG}/llama-3.2-1B/QLoRA/llama3_2_qat_lora.pte`;
13
+ const LLAMA3_2_1B_SPINQUANT_MODEL = `${URL_PREFIX}-llama-3.2/${VERSION_TAG}/llama-3.2-1B/spinquant/llama3_2_spinquant.pte`;
14
+ const LLAMA3_2_TOKENIZER = `${URL_PREFIX}-llama-3.2/${VERSION_TAG}/tokenizer.json`;
15
+ const LLAMA3_2_TOKENIZER_CONFIG = `${URL_PREFIX}-llama-3.2/${VERSION_TAG}/tokenizer_config.json`;
16
+ export const LLAMA3_2_3B = {
17
+ modelSource: LLAMA3_2_3B_MODEL,
18
+ tokenizerSource: LLAMA3_2_TOKENIZER,
19
+ tokenizerConfigSource: LLAMA3_2_TOKENIZER_CONFIG,
20
+ };
21
+ export const LLAMA3_2_3B_QLORA = {
22
+ modelSource: LLAMA3_2_3B_QLORA_MODEL,
23
+ tokenizerSource: LLAMA3_2_TOKENIZER,
24
+ tokenizerConfigSource: LLAMA3_2_TOKENIZER_CONFIG,
25
+ };
26
+ export const LLAMA3_2_3B_SPINQUANT = {
27
+ modelSource: LLAMA3_2_3B_SPINQUANT_MODEL,
28
+ tokenizerSource: LLAMA3_2_TOKENIZER,
29
+ tokenizerConfigSource: LLAMA3_2_TOKENIZER_CONFIG,
30
+ };
31
+ export const LLAMA3_2_1B = {
32
+ modelSource: LLAMA3_2_1B_MODEL,
33
+ tokenizerSource: LLAMA3_2_TOKENIZER,
34
+ tokenizerConfigSource: LLAMA3_2_TOKENIZER_CONFIG,
35
+ };
36
+ export const LLAMA3_2_1B_QLORA = {
37
+ modelSource: LLAMA3_2_1B_QLORA_MODEL,
38
+ tokenizerSource: LLAMA3_2_TOKENIZER,
39
+ tokenizerConfigSource: LLAMA3_2_TOKENIZER_CONFIG,
40
+ };
41
+ export const LLAMA3_2_1B_SPINQUANT = {
42
+ modelSource: LLAMA3_2_1B_SPINQUANT_MODEL,
43
+ tokenizerSource: LLAMA3_2_TOKENIZER,
44
+ tokenizerConfigSource: LLAMA3_2_TOKENIZER_CONFIG,
45
+ };
46
+ // QWEN 3
47
+ const QWEN3_0_6B_MODEL = `${URL_PREFIX}-qwen-3/${VERSION_TAG}/qwen-3-0.6B/original/qwen3_0_6b_bf16.pte`;
48
+ const QWEN3_0_6B_QUANTIZED_MODEL = `${URL_PREFIX}-qwen-3/${VERSION_TAG}/qwen-3-0.6B/quantized/qwen3_0_6b_8da4w.pte`;
49
+ const QWEN3_1_7B_MODEL = `${URL_PREFIX}-qwen-3/${VERSION_TAG}/qwen-3-1.7B/original/qwen3_1_7b_bf16.pte`;
50
+ const QWEN3_1_7B_QUANTIZED_MODEL = `${URL_PREFIX}-qwen-3/${VERSION_TAG}/qwen-3-1.7B/quantized/qwen3_1_7b_8da4w.pte`;
51
+ const QWEN3_4B_MODEL = `${URL_PREFIX}-qwen-3/${VERSION_TAG}/qwen-3-4B/original/qwen3_4b_bf16.pte`;
52
+ const QWEN3_4B_QUANTIZED_MODEL = `${URL_PREFIX}-qwen-3/${VERSION_TAG}/qwen-3-4B/quantized/qwen3_4b_8da4w.pte`;
53
+ const QWEN3_TOKENIZER = `${URL_PREFIX}-qwen-3/${VERSION_TAG}/tokenizer.json`;
54
+ const QWEN3_TOKENIZER_CONFIG = `${URL_PREFIX}-qwen-3/${VERSION_TAG}/tokenizer_config.json`;
55
+ export const QWEN3_0_6B = {
56
+ modelSource: QWEN3_0_6B_MODEL,
57
+ tokenizerSource: QWEN3_TOKENIZER,
58
+ tokenizerConfigSource: QWEN3_TOKENIZER_CONFIG,
59
+ };
60
+ export const QWEN3_0_6B_QUANTIZED = {
61
+ modelSource: QWEN3_0_6B_QUANTIZED_MODEL,
62
+ tokenizerSource: QWEN3_TOKENIZER,
63
+ tokenizerConfigSource: QWEN3_TOKENIZER_CONFIG,
64
+ };
65
+ export const QWEN3_1_7B = {
66
+ modelSource: QWEN3_1_7B_MODEL,
67
+ tokenizerSource: QWEN3_TOKENIZER,
68
+ tokenizerConfigSource: QWEN3_TOKENIZER_CONFIG,
69
+ };
70
+ export const QWEN3_1_7B_QUANTIZED = {
71
+ modelSource: QWEN3_1_7B_QUANTIZED_MODEL,
72
+ tokenizerSource: QWEN3_TOKENIZER,
73
+ tokenizerConfigSource: QWEN3_TOKENIZER_CONFIG,
74
+ };
75
+ export const QWEN3_4B = {
76
+ modelSource: QWEN3_4B_MODEL,
77
+ tokenizerSource: QWEN3_TOKENIZER,
78
+ tokenizerConfigSource: QWEN3_TOKENIZER_CONFIG,
79
+ };
80
+ export const QWEN3_4B_QUANTIZED = {
81
+ modelSource: QWEN3_4B_QUANTIZED_MODEL,
82
+ tokenizerSource: QWEN3_TOKENIZER,
83
+ tokenizerConfigSource: QWEN3_TOKENIZER_CONFIG,
84
+ };
85
+ // HAMMER 2.1
86
+ const HAMMER2_1_0_5B_MODEL = `${URL_PREFIX}-hammer-2.1/${VERSION_TAG}/hammer-2.1-0.5B/original/hammer2_1_0_5B_bf16.pte`;
87
+ const HAMMER2_1_0_5B_QUANTIZED_MODEL = `${URL_PREFIX}-hammer-2.1/${VERSION_TAG}/hammer-2.1-0.5B/quantized/hammer2_1_0_5B_8da4w.pte`;
88
+ const HAMMER2_1_1_5B_MODEL = `${URL_PREFIX}-hammer-2.1/${VERSION_TAG}/hammer-2.1-1.5B/original/hammer2_1_1_5B_bf16.pte`;
89
+ const HAMMER2_1_1_5B_QUANTIZED_MODEL = `${URL_PREFIX}-hammer-2.1/${VERSION_TAG}/hammer-2.1-1.5B/quantized/hammer2_1_1_5B_8da4w.pte`;
90
+ const HAMMER2_1_3B_MODEL = `${URL_PREFIX}-hammer-2.1/${VERSION_TAG}/hammer-2.1-3B/original/hammer2_1_3B_bf16.pte`;
91
+ const HAMMER2_1_3B_QUANTIZED_MODEL = `${URL_PREFIX}-hammer-2.1/${VERSION_TAG}/hammer-2.1-3B/quantized/hammer2_1_3B_8da4w.pte`;
92
+ const HAMMER2_1_TOKENIZER = `${URL_PREFIX}-hammer-2.1/${VERSION_TAG}/tokenizer.json`;
93
+ const HAMMER2_1_TOKENIZER_CONFIG = `${URL_PREFIX}-hammer-2.1/${VERSION_TAG}/tokenizer_config.json`;
94
+ export const HAMMER2_1_0_5B = {
95
+ modelSource: HAMMER2_1_0_5B_MODEL,
96
+ tokenizerSource: HAMMER2_1_TOKENIZER,
97
+ tokenizerConfigSource: HAMMER2_1_TOKENIZER_CONFIG,
98
+ };
99
+ export const HAMMER2_1_0_5B_QUANTIZED = {
100
+ modelSource: HAMMER2_1_0_5B_QUANTIZED_MODEL,
101
+ tokenizerSource: HAMMER2_1_TOKENIZER,
102
+ tokenizerConfigSource: HAMMER2_1_TOKENIZER_CONFIG,
103
+ };
104
+ export const HAMMER2_1_1_5B = {
105
+ modelSource: HAMMER2_1_1_5B_MODEL,
106
+ tokenizerSource: HAMMER2_1_TOKENIZER,
107
+ tokenizerConfigSource: HAMMER2_1_TOKENIZER_CONFIG,
108
+ };
109
+ export const HAMMER2_1_1_5B_QUANTIZED = {
110
+ modelSource: HAMMER2_1_1_5B_QUANTIZED_MODEL,
111
+ tokenizerSource: HAMMER2_1_TOKENIZER,
112
+ tokenizerConfigSource: HAMMER2_1_TOKENIZER_CONFIG,
113
+ };
114
+ export const HAMMER2_1_3B = {
115
+ modelSource: HAMMER2_1_3B_MODEL,
116
+ tokenizerSource: HAMMER2_1_TOKENIZER,
117
+ tokenizerConfigSource: HAMMER2_1_TOKENIZER_CONFIG,
118
+ };
119
+ export const HAMMER2_1_3B_QUANTIZED = {
120
+ modelSource: HAMMER2_1_3B_QUANTIZED_MODEL,
121
+ tokenizerSource: HAMMER2_1_TOKENIZER,
122
+ tokenizerConfigSource: HAMMER2_1_TOKENIZER_CONFIG,
123
+ };
124
+ // SMOLLM2
125
+ const SMOLLM2_1_135M_MODEL = `${URL_PREFIX}-smolLm-2/${VERSION_TAG}/smolLm-2-135M/original/smolLm2_135M_bf16.pte`;
126
+ const SMOLLM2_1_135M_QUANTIZED_MODEL = `${URL_PREFIX}-smolLm-2/${VERSION_TAG}/smolLm-2-135M/quantized/smolLm2_135M_8da4w.pte`;
127
+ const SMOLLM2_1_360M_MODEL = `${URL_PREFIX}-smolLm-2/${VERSION_TAG}/smolLm-2-360M/original/smolLm2_360M_bf16.pte`;
128
+ const SMOLLM2_1_360M_QUANTIZED_MODEL = `${URL_PREFIX}-smolLm-2/${VERSION_TAG}/smolLm-2-360M/quantized/smolLm2_360M_8da4w.pte`;
129
+ const SMOLLM2_1_1_7B_MODEL = `${URL_PREFIX}-smolLm-2/${VERSION_TAG}/smolLm-2-1.7B/original/smolLm2_1_7B_bf16.pte`;
130
+ const SMOLLM2_1_1_7B_QUANTIZED_MODEL = `${URL_PREFIX}-smolLm-2/${VERSION_TAG}/smolLm-2-1.7B/quantized/smolLm2_1_7B_8da4w.pte`;
131
+ const SMOLLM2_1_TOKENIZER = `${URL_PREFIX}-smolLm-2/${VERSION_TAG}/tokenizer.json`;
132
+ const SMOLLM2_1_TOKENIZER_CONFIG = `${URL_PREFIX}-smolLm-2/${VERSION_TAG}/tokenizer_config.json`;
133
+ export const SMOLLM2_1_135M = {
134
+ modelSource: SMOLLM2_1_135M_MODEL,
135
+ tokenizerSource: SMOLLM2_1_TOKENIZER,
136
+ tokenizerConfigSource: SMOLLM2_1_TOKENIZER_CONFIG,
137
+ };
138
+ export const SMOLLM2_1_135M_QUANTIZED = {
139
+ modelSource: SMOLLM2_1_135M_QUANTIZED_MODEL,
140
+ tokenizerSource: SMOLLM2_1_TOKENIZER,
141
+ tokenizerConfigSource: SMOLLM2_1_TOKENIZER_CONFIG,
142
+ };
143
+ export const SMOLLM2_1_360M = {
144
+ modelSource: SMOLLM2_1_360M_MODEL,
145
+ tokenizerSource: SMOLLM2_1_TOKENIZER,
146
+ tokenizerConfigSource: SMOLLM2_1_TOKENIZER_CONFIG,
147
+ };
148
+ export const SMOLLM2_1_360M_QUANTIZED = {
149
+ modelSource: SMOLLM2_1_360M_QUANTIZED_MODEL,
150
+ tokenizerSource: SMOLLM2_1_TOKENIZER,
151
+ tokenizerConfigSource: SMOLLM2_1_TOKENIZER_CONFIG,
152
+ };
153
+ export const SMOLLM2_1_1_7B = {
154
+ modelSource: SMOLLM2_1_1_7B_MODEL,
155
+ tokenizerSource: SMOLLM2_1_TOKENIZER,
156
+ tokenizerConfigSource: SMOLLM2_1_TOKENIZER_CONFIG,
157
+ };
158
+ export const SMOLLM2_1_1_7B_QUANTIZED = {
159
+ modelSource: SMOLLM2_1_1_7B_QUANTIZED_MODEL,
160
+ tokenizerSource: SMOLLM2_1_TOKENIZER,
161
+ tokenizerConfigSource: SMOLLM2_1_TOKENIZER_CONFIG,
162
+ };
163
+ // QWEN 2.5
164
+ const QWEN2_5_0_5B_MODEL = `${URL_PREFIX}-qwen-2.5/${VERSION_TAG}/qwen-2.5-0.5B/original/qwen2_5_0_5b_bf16.pte`;
165
+ const QWEN2_5_0_5B_QUANTIZED_MODEL = `${URL_PREFIX}-qwen-2.5/${VERSION_TAG}/qwen-2.5-0.5B/quantized/qwen2_5_0_5b_8da4w.pte`;
166
+ const QWEN2_5_1_5B_MODEL = `${URL_PREFIX}-qwen-2.5/${VERSION_TAG}/qwen-2.5-1.5B/original/qwen2_5_1_5b_bf16.pte`;
167
+ const QWEN2_5_1_5B_QUANTIZED_MODEL = `${URL_PREFIX}-qwen-2.5/${VERSION_TAG}/qwen-2.5-1.5B/quantized/qwen2_5_1_5b_8da4w.pte`;
168
+ const QWEN2_5_3B_MODEL = `${URL_PREFIX}-qwen-2.5/${VERSION_TAG}/qwen-2.5-3B/original/qwen2_5_3b_bf16.pte`;
169
+ const QWEN2_5_3B_QUANTIZED_MODEL = `${URL_PREFIX}-qwen-2.5/${VERSION_TAG}/qwen-2.5-3B/quantized/qwen2_5_3b_8da4w.pte`;
170
+ const QWEN2_5_TOKENIZER = `${URL_PREFIX}-qwen-2.5/${VERSION_TAG}/tokenizer.json`;
171
+ const QWEN2_5_TOKENIZER_CONFIG = `${URL_PREFIX}-qwen-2.5/${VERSION_TAG}/tokenizer_config.json`;
172
+ export const QWEN2_5_0_5B = {
173
+ modelSource: QWEN2_5_0_5B_MODEL,
174
+ tokenizerSource: QWEN2_5_TOKENIZER,
175
+ tokenizerConfigSource: QWEN2_5_TOKENIZER_CONFIG,
176
+ };
177
+ export const QWEN2_5_0_5B_QUANTIZED = {
178
+ modelSource: QWEN2_5_0_5B_QUANTIZED_MODEL,
179
+ tokenizerSource: QWEN2_5_TOKENIZER,
180
+ tokenizerConfigSource: QWEN2_5_TOKENIZER_CONFIG,
181
+ };
182
+ export const QWEN2_5_1_5B = {
183
+ modelSource: QWEN2_5_1_5B_MODEL,
184
+ tokenizerSource: QWEN2_5_TOKENIZER,
185
+ tokenizerConfigSource: QWEN2_5_TOKENIZER_CONFIG,
186
+ };
187
+ export const QWEN2_5_1_5B_QUANTIZED = {
188
+ modelSource: QWEN2_5_1_5B_QUANTIZED_MODEL,
189
+ tokenizerSource: QWEN2_5_TOKENIZER,
190
+ tokenizerConfigSource: QWEN2_5_TOKENIZER_CONFIG,
191
+ };
192
+ export const QWEN2_5_3B = {
193
+ modelSource: QWEN2_5_3B_MODEL,
194
+ tokenizerSource: QWEN2_5_TOKENIZER,
195
+ tokenizerConfigSource: QWEN2_5_TOKENIZER_CONFIG,
196
+ };
197
+ export const QWEN2_5_3B_QUANTIZED = {
198
+ modelSource: QWEN2_5_3B_QUANTIZED_MODEL,
199
+ tokenizerSource: QWEN2_5_TOKENIZER,
200
+ tokenizerConfigSource: QWEN2_5_TOKENIZER_CONFIG,
201
+ };
202
+ // PHI 4
203
+ const PHI_4_MINI_4B_MODEL = `${URL_PREFIX}-phi-4-mini/${VERSION_TAG}/original/phi-4-mini_bf16.pte`;
204
+ const PHI_4_MINI_4B_QUANTIZED_MODEL = `${URL_PREFIX}-phi-4-mini/${VERSION_TAG}/quantized/phi-4-mini_8da4w.pte`;
205
+ const PHI_4_MINI_TOKENIZER = `${URL_PREFIX}-phi-4-mini/${VERSION_TAG}/tokenizer.json`;
206
+ const PHI_4_MINI_TOKENIZER_CONFIG = `${URL_PREFIX}-phi-4-mini/${VERSION_TAG}/tokenizer_config.json`;
207
+ export const PHI_4_MINI_4B = {
208
+ modelSource: PHI_4_MINI_4B_MODEL,
209
+ tokenizerSource: PHI_4_MINI_TOKENIZER,
210
+ tokenizerConfigSource: PHI_4_MINI_TOKENIZER_CONFIG,
211
+ };
212
+ export const PHI_4_MINI_4B_QUANTIZED = {
213
+ modelSource: PHI_4_MINI_4B_QUANTIZED_MODEL,
214
+ tokenizerSource: PHI_4_MINI_TOKENIZER,
215
+ tokenizerConfigSource: PHI_4_MINI_TOKENIZER_CONFIG,
216
+ };
217
+ // Classification
218
+ const EFFICIENTNET_V2_S_MODEL = Platform.OS === `ios`
219
+ ? `${URL_PREFIX}-efficientnet-v2-s/${VERSION_TAG}/coreml/efficientnet_v2_s_coreml_all.pte`
220
+ : `${URL_PREFIX}-efficientnet-v2-s/${VERSION_TAG}/xnnpack/efficientnet_v2_s_xnnpack.pte`;
221
+ export const EFFICIENTNET_V2_S = {
222
+ modelSource: EFFICIENTNET_V2_S_MODEL,
223
+ };
224
+ // Object detection
225
+ const SSDLITE_320_MOBILENET_V3_LARGE_MODEL = `${URL_PREFIX}-ssdlite320-mobilenet-v3-large/${VERSION_TAG}/ssdlite320-mobilenetv3-large.pte`;
226
+ export const SSDLITE_320_MOBILENET_V3_LARGE = {
227
+ modelSource: SSDLITE_320_MOBILENET_V3_LARGE_MODEL,
228
+ };
229
+ // Style transfer
230
+ const STYLE_TRANSFER_CANDY_MODEL = Platform.OS === `ios`
231
+ ? `${URL_PREFIX}-style-transfer-candy/${VERSION_TAG}/coreml/style_transfer_candy_coreml.pte`
232
+ : `${URL_PREFIX}-style-transfer-candy/${VERSION_TAG}/xnnpack/style_transfer_candy_xnnpack.pte`;
233
+ const STYLE_TRANSFER_MOSAIC_MODEL = Platform.OS === `ios`
234
+ ? `${URL_PREFIX}-style-transfer-mosaic/${VERSION_TAG}/coreml/style_transfer_mosaic_coreml.pte`
235
+ : `${URL_PREFIX}-style-transfer-mosaic/${VERSION_TAG}/xnnpack/style_transfer_mosaic_xnnpack.pte`;
236
+ const STYLE_TRANSFER_RAIN_PRINCESS_MODEL = Platform.OS === `ios`
237
+ ? `${URL_PREFIX}-style-transfer-rain-princess/${VERSION_TAG}/coreml/style_transfer_rain_princess_coreml.pte`
238
+ : `${URL_PREFIX}-style-transfer-rain-princess/${VERSION_TAG}/xnnpack/style_transfer_rain_princess_xnnpack.pte`;
239
+ const STYLE_TRANSFER_UDNIE_MODEL = Platform.OS === `ios`
240
+ ? `${URL_PREFIX}-style-transfer-udnie/${VERSION_TAG}/coreml/style_transfer_udnie_coreml.pte`
241
+ : `${URL_PREFIX}-style-transfer-udnie/${VERSION_TAG}/xnnpack/style_transfer_udnie_xnnpack.pte`;
242
+ export const STYLE_TRANSFER_CANDY = {
243
+ modelSource: STYLE_TRANSFER_CANDY_MODEL,
244
+ };
245
+ export const STYLE_TRANSFER_MOSAIC = {
246
+ modelSource: STYLE_TRANSFER_MOSAIC_MODEL,
247
+ };
248
+ export const STYLE_TRANSFER_RAIN_PRINCESS = {
249
+ modelSource: STYLE_TRANSFER_RAIN_PRINCESS_MODEL,
250
+ };
251
+ export const STYLE_TRANSFER_UDNIE = {
252
+ modelSource: STYLE_TRANSFER_UDNIE_MODEL,
253
+ };
254
+ // S2T
255
+ const MOONSHINE_TINY_DECODER_MODEL = `${URL_PREFIX}-moonshine-tiny/${VERSION_TAG}/xnnpack/moonshine_tiny_xnnpack_decoder.pte`;
256
+ const MOONSHINE_TINY_ENCODER_MODEL = `${URL_PREFIX}-moonshine-tiny/${VERSION_TAG}/xnnpack/moonshine_tiny_xnnpack_encoder.pte`;
257
+ const MOONSHINE_TOKENIZER = `${URL_PREFIX}-moonshine-tiny/${VERSION_TAG}/moonshine_tiny_tokenizer.json`;
258
+ const WHISPER_TOKENIZER = `${URL_PREFIX}-whisper-tiny.en/${VERSION_TAG}/whisper_tokenizer.json`;
259
+ const WHISPER_TINY_DECODER_MODEL = `${URL_PREFIX}-whisper-tiny.en/${VERSION_TAG}/xnnpack/whisper_tiny_en_xnnpack_decoder.pte`;
260
+ const WHISPER_TINY_ENCODER_MODEL = `${URL_PREFIX}-whisper-tiny.en/${VERSION_TAG}/xnnpack/whisper_tiny_en_xnnpack_encoder.pte`;
261
+ const WHISPER_TINY_MULTILINGUAL_ENCODER_MODEL = `${URL_PREFIX}-whisper-tiny/${VERSION_TAG}/xnnpack/xnnpack_whisper_encoder.pte`;
262
+ const WHISPER_TINY_MULTILINGUAL_DECODER_MODEL = `${URL_PREFIX}-whisper-tiny/${VERSION_TAG}/xnnpack/xnnpack_whisper_decoder.pte`;
263
+ const WHISPER_TINY_MULTILINGUAL_TOKENIZER = `${URL_PREFIX}-whisper-tiny/${VERSION_TAG}/tokenizer.json`;
264
+ export const MOONSHINE_TINY = {
265
+ modelName: AvailableModels.MOONSHINE,
266
+ decoderSource: MOONSHINE_TINY_DECODER_MODEL,
267
+ encoderSource: MOONSHINE_TINY_ENCODER_MODEL,
268
+ tokenizerSource: MOONSHINE_TOKENIZER,
269
+ };
270
+ export const WHISPER_TINY = {
271
+ modelName: AvailableModels.WHISPER,
272
+ decoderSource: WHISPER_TINY_DECODER_MODEL,
273
+ encoderSource: WHISPER_TINY_ENCODER_MODEL,
274
+ tokenizerSource: WHISPER_TOKENIZER,
275
+ };
276
+ export const WHISPER_TINY_MULTILINGUAL = {
277
+ modelName: AvailableModels.WHISPER_MULTILINGUAL,
278
+ decoderSource: WHISPER_TINY_MULTILINGUAL_DECODER_MODEL,
279
+ encoderSource: WHISPER_TINY_MULTILINGUAL_ENCODER_MODEL,
280
+ tokenizerSource: WHISPER_TINY_MULTILINGUAL_TOKENIZER,
281
+ };
282
+ // Image segmentation
283
+ const DEEPLAB_V3_RESNET50_MODEL = `${URL_PREFIX}-deeplab-v3/${VERSION_TAG}/xnnpack/deeplabV3_xnnpack_fp32.pte`;
284
+ export const DEEPLAB_V3_RESNET50 = {
285
+ modelSource: DEEPLAB_V3_RESNET50_MODEL,
286
+ };
287
+ // Image Embeddings
288
+ const CLIP_VIT_BASE_PATCH32_IMAGE_MODEL = `${URL_PREFIX}-clip-vit-base-patch32/${NEXT_VERSION_TAG}/clip-vit-base-patch32-vision_xnnpack.pte`;
289
+ export const CLIP_VIT_BASE_PATCH32_IMAGE = {
290
+ modelSource: CLIP_VIT_BASE_PATCH32_IMAGE_MODEL,
291
+ };
292
+ // Text Embeddings
293
+ const ALL_MINILM_L6_V2_MODEL = `${URL_PREFIX}-all-MiniLM-L6-v2/${NEXT_VERSION_TAG}/all-MiniLM-L6-v2_xnnpack.pte`;
294
+ const ALL_MINILM_L6_V2_TOKENIZER = `${URL_PREFIX}-all-MiniLM-L6-v2/${NEXT_VERSION_TAG}/tokenizer.json`;
295
+ const ALL_MPNET_BASE_V2_MODEL = `${URL_PREFIX}-all-mpnet-base-v2/${NEXT_VERSION_TAG}/all-mpnet-base-v2_xnnpack.pte`;
296
+ const ALL_MPNET_BASE_V2_TOKENIZER = `${URL_PREFIX}-all-mpnet-base-v2/${NEXT_VERSION_TAG}/tokenizer.json`;
297
+ const MULTI_QA_MINILM_L6_COS_V1_MODEL = `${URL_PREFIX}-multi-qa-MiniLM-L6-cos-v1/${NEXT_VERSION_TAG}/multi-qa-MiniLM-L6-cos-v1_xnnpack.pte`;
298
+ const MULTI_QA_MINILM_L6_COS_V1_TOKENIZER = `${URL_PREFIX}-multi-qa-MiniLM-L6-cos-v1/${NEXT_VERSION_TAG}/tokenizer.json`;
299
+ const MULTI_QA_MPNET_BASE_DOT_V1_MODEL = `${URL_PREFIX}-multi-qa-mpnet-base-dot-v1/${NEXT_VERSION_TAG}/multi-qa-mpnet-base-dot-v1_xnnpack.pte`;
300
+ const MULTI_QA_MPNET_BASE_DOT_V1_TOKENIZER = `${URL_PREFIX}-multi-qa-mpnet-base-dot-v1/${NEXT_VERSION_TAG}/tokenizer.json`;
301
+ const CLIP_VIT_BASE_PATCH32_TEXT_MODEL = `${URL_PREFIX}-clip-vit-base-patch32/${NEXT_VERSION_TAG}/clip-vit-base-patch32-text_xnnpack.pte`;
302
+ const CLIP_VIT_BASE_PATCH32_TEXT_TOKENIZER = `${URL_PREFIX}-clip-vit-base-patch32/${NEXT_VERSION_TAG}/tokenizer.json`;
303
+ export const ALL_MINILM_L6_V2 = {
304
+ modelSource: ALL_MINILM_L6_V2_MODEL,
305
+ tokenizerSource: ALL_MINILM_L6_V2_TOKENIZER,
306
+ };
307
+ export const ALL_MPNET_BASE_V2 = {
308
+ modelSource: ALL_MPNET_BASE_V2_MODEL,
309
+ tokenizerSource: ALL_MPNET_BASE_V2_TOKENIZER,
310
+ };
311
+ export const MULTI_QA_MINILM_L6_COS_V1 = {
312
+ modelSource: MULTI_QA_MINILM_L6_COS_V1_MODEL,
313
+ tokenizerSource: MULTI_QA_MINILM_L6_COS_V1_TOKENIZER,
314
+ };
315
+ export const MULTI_QA_MPNET_BASE_DOT_V1 = {
316
+ modelSource: MULTI_QA_MPNET_BASE_DOT_V1_MODEL,
317
+ tokenizerSource: MULTI_QA_MPNET_BASE_DOT_V1_TOKENIZER,
318
+ };
319
+ export const CLIP_VIT_BASE_PATCH32_TEXT = {
320
+ modelSource: CLIP_VIT_BASE_PATCH32_TEXT_MODEL,
321
+ tokenizerSource: CLIP_VIT_BASE_PATCH32_TEXT_TOKENIZER,
322
+ };