Skip to content

Commit 7086044

Browse files
msluszniakmsluszniak
committed
refactor(constants): hoist multi-backend URLs to modelUrls.ts
modelRegistry.ts duplicated the same `${URL_PREFIX}-…/${VERSION_TAG}/…` strings that modelUrls.ts already had inline in each Platform.OS branch. Hoist a single set of per-backend URL constants into modelUrls.ts and have both consumers reference them, so each URL string lives in exactly one place. - Add per-backend exports for efficientnet-v2-s, ssdlite320-mobilenet-v3- large, rfdetr-nano-detector, rfdetr-nano-segmentation, fast-sam {s,x}, distiluse-base-multilingual-cased-v2. - Add `styleTransferUrls(display, slug)` helper for the 4 style-transfer styles; the registry's `styleTransferVariants` now consumes it. - Drop the now-unused `URL_PREFIX, VERSION_TAG` import from modelRegistry.ts. Addresses #1148 (comment)
1 parent d849b0a commit 7086044

2 files changed

Lines changed: 81 additions & 68 deletions

File tree

‎packages/react-native-executorch/src/constants/modelRegistry.ts‎

Lines changed: 23 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
import { Platform } from 'react-native';
22
import * as M from './modelUrls';
3-
import { URL_PREFIX, VERSION_TAG } from './versions';
43

54
/**
65
* Backend options accepted by `MODEL_REGISTRY` accessors.
@@ -150,21 +149,21 @@ const EFFICIENTNET_V2_S_VARIANTS = {
150149
xnnpack: {
151150
base: {
152151
modelName: 'efficientnet-v2-s' as const,
153-
modelSource: `${URL_PREFIX}-efficientnet-v2-s/${VERSION_TAG}/xnnpack/efficientnet_v2_s_xnnpack_fp32.pte`,
152+
modelSource: M.EFFICIENTNET_V2_S_XNNPACK_FP32_MODEL,
154153
},
155154
quant: {
156155
modelName: 'efficientnet-v2-s-quantized' as const,
157-
modelSource: `${URL_PREFIX}-efficientnet-v2-s/${VERSION_TAG}/xnnpack/efficientnet_v2_s_xnnpack_int8.pte`,
156+
modelSource: M.EFFICIENTNET_V2_S_XNNPACK_INT8_MODEL,
158157
},
159158
},
160159
coreml: {
161160
base: {
162161
modelName: 'efficientnet-v2-s' as const,
163-
modelSource: `${URL_PREFIX}-efficientnet-v2-s/${VERSION_TAG}/coreml/efficientnet_v2_s_coreml_fp32.pte`,
162+
modelSource: M.EFFICIENTNET_V2_S_COREML_FP32_MODEL,
164163
},
165164
quant: {
166165
modelName: 'efficientnet-v2-s-quantized' as const,
167-
modelSource: `${URL_PREFIX}-efficientnet-v2-s/${VERSION_TAG}/coreml/efficientnet_v2_s_coreml_fp16.pte`,
166+
modelSource: M.EFFICIENTNET_V2_S_COREML_FP16_MODEL,
168167
},
169168
},
170169
};
@@ -173,13 +172,13 @@ const SSDLITE_320_MOBILENET_V3_LARGE_VARIANTS = {
173172
xnnpack: {
174173
base: {
175174
modelName: 'ssdlite-320-mobilenet-v3-large' as const,
176-
modelSource: `${URL_PREFIX}-ssdlite320-mobilenet-v3-large/${VERSION_TAG}/xnnpack/ssdlite320_mobilenet_v3_large_xnnpack_fp32.pte`,
175+
modelSource: M.SSDLITE_320_MOBILENET_V3_LARGE_XNNPACK_FP32_MODEL,
177176
},
178177
},
179178
coreml: {
180179
base: {
181180
modelName: 'ssdlite-320-mobilenet-v3-large' as const,
182-
modelSource: `${URL_PREFIX}-ssdlite320-mobilenet-v3-large/${VERSION_TAG}/coreml/ssdlite320_mobilenet_v3_large_coreml_fp16.pte`,
181+
modelSource: M.SSDLITE_320_MOBILENET_V3_LARGE_COREML_FP16_MODEL,
183182
},
184183
},
185184
};
@@ -188,13 +187,13 @@ const RF_DETR_NANO_VARIANTS = {
188187
xnnpack: {
189188
base: {
190189
modelName: 'rf-detr-nano' as const,
191-
modelSource: `${URL_PREFIX}-rfdetr-nano-detector/${VERSION_TAG}/xnnpack/rfdetr_nano_xnnpack_fp32.pte`,
190+
modelSource: M.RF_DETR_NANO_XNNPACK_FP32_MODEL,
192191
},
193192
},
194193
coreml: {
195194
base: {
196195
modelName: 'rf-detr-nano' as const,
197-
modelSource: `${URL_PREFIX}-rfdetr-nano-detector/${VERSION_TAG}/coreml/rfdetr_nano_coreml_int8.pte`,
196+
modelSource: M.RF_DETR_NANO_COREML_INT8_MODEL,
198197
},
199198
},
200199
};
@@ -203,13 +202,13 @@ const RF_DETR_NANO_SEG_VARIANTS = {
203202
xnnpack: {
204203
base: {
205204
modelName: 'rfdetr-nano-seg' as const,
206-
modelSource: `${URL_PREFIX}-rfdetr-nano-segmentation/${VERSION_TAG}/xnnpack/rfdetr_nano_xnnpack_fp32.pte`,
205+
modelSource: M.RF_DETR_NANO_SEG_XNNPACK_FP32_MODEL,
207206
},
208207
},
209208
coreml: {
210209
base: {
211210
modelName: 'rfdetr-nano-seg' as const,
212-
modelSource: `${URL_PREFIX}-rfdetr-nano-segmentation/${VERSION_TAG}/coreml/rfdetr_nano_coreml_int8.pte`,
211+
modelSource: M.RF_DETR_NANO_SEG_COREML_INT8_MODEL,
213212
},
214213
},
215214
};
@@ -218,13 +217,13 @@ const FASTSAM_S_VARIANTS = {
218217
xnnpack: {
219218
base: {
220219
modelName: 'fastsam-s' as const,
221-
modelSource: `${URL_PREFIX}-fast-sam/${VERSION_TAG}/s/xnnpack/fast_sam_s_xnnpack_fp32.pte`,
220+
modelSource: M.FASTSAM_S_XNNPACK_FP32_MODEL,
222221
},
223222
},
224223
coreml: {
225224
base: {
226225
modelName: 'fastsam-s' as const,
227-
modelSource: `${URL_PREFIX}-fast-sam/${VERSION_TAG}/s/coreml/fast_sam_s_coreml_fp16.pte`,
226+
modelSource: M.FASTSAM_S_COREML_FP16_MODEL,
228227
},
229228
},
230229
};
@@ -233,13 +232,13 @@ const FASTSAM_X_VARIANTS = {
233232
xnnpack: {
234233
base: {
235234
modelName: 'fastsam-x' as const,
236-
modelSource: `${URL_PREFIX}-fast-sam/${VERSION_TAG}/x/xnnpack/fast_sam_x_xnnpack_fp32.pte`,
235+
modelSource: M.FASTSAM_X_XNNPACK_FP32_MODEL,
237236
},
238237
},
239238
coreml: {
240239
base: {
241240
modelName: 'fastsam-x' as const,
242-
modelSource: `${URL_PREFIX}-fast-sam/${VERSION_TAG}/x/coreml/fast_sam_x_coreml_fp16.pte`,
241+
modelSource: M.FASTSAM_X_COREML_FP16_MODEL,
243242
},
244243
},
245244
};
@@ -248,18 +247,19 @@ function styleTransferVariants<
248247
const Display extends string,
249248
const Slug extends string,
250249
>(display: Display, slug: Slug) {
250+
const urls = M.styleTransferUrls(display, slug);
251251
return {
252252
xnnpack: {
253253
base: {
254254
modelName: `style-transfer-${display}`,
255-
modelSource: `${URL_PREFIX}-style-transfer-${display}/${VERSION_TAG}/xnnpack/style_transfer_${slug}_xnnpack_fp32.pte`,
255+
modelSource: urls.xnnpackBase,
256256
} as {
257257
modelName: `style-transfer-${Display}`;
258258
modelSource: string;
259259
},
260260
quant: {
261261
modelName: `style-transfer-${display}-quantized`,
262-
modelSource: `${URL_PREFIX}-style-transfer-${display}/${VERSION_TAG}/xnnpack/style_transfer_${slug}_xnnpack_int8.pte`,
262+
modelSource: urls.xnnpackQuant,
263263
} as {
264264
modelName: `style-transfer-${Display}-quantized`;
265265
modelSource: string;
@@ -268,14 +268,14 @@ function styleTransferVariants<
268268
coreml: {
269269
base: {
270270
modelName: `style-transfer-${display}`,
271-
modelSource: `${URL_PREFIX}-style-transfer-${display}/${VERSION_TAG}/coreml/style_transfer_${slug}_coreml_fp32.pte`,
271+
modelSource: urls.coremlBase,
272272
} as {
273273
modelName: `style-transfer-${Display}`;
274274
modelSource: string;
275275
},
276276
quant: {
277277
modelName: `style-transfer-${display}-quantized`,
278-
modelSource: `${URL_PREFIX}-style-transfer-${display}/${VERSION_TAG}/coreml/style_transfer_${slug}_coreml_fp16.pte`,
278+
modelSource: urls.coremlQuant,
279279
} as {
280280
modelName: `style-transfer-${Display}-quantized`;
281281
modelSource: string;
@@ -295,21 +295,19 @@ const STYLE_TRANSFER_RAIN_PRINCESS_VARIANTS = styleTransferVariants(
295295
);
296296
const STYLE_TRANSFER_UDNIE_VARIANTS = styleTransferVariants('udnie', 'udnie');
297297

298-
const DISTILUSE_TOKENIZER = `${URL_PREFIX}-distiluse-base-multilingual-cased-v2/${VERSION_TAG}/tokenizer.json`;
299-
300298
const DISTILUSE_BASE_MULTILINGUAL_CASED_V2_VARIANTS = {
301299
xnnpack: {
302300
base: {
303301
modelName: 'distiluse-base-multilingual-cased-v2-8da4w' as const,
304-
modelSource: `${URL_PREFIX}-distiluse-base-multilingual-cased-v2/${VERSION_TAG}/xnnpack/distiluse_base_multilingual_cased_v2_xnnpack_8da4w.pte`,
305-
tokenizerSource: DISTILUSE_TOKENIZER,
302+
modelSource: M.DISTILUSE_BASE_MULTILINGUAL_CASED_V2_8DA4W_MODEL,
303+
tokenizerSource: M.DISTILUSE_BASE_MULTILINGUAL_CASED_V2_TOKENIZER,
306304
},
307305
},
308306
coreml: {
309307
base: {
310308
modelName: 'distiluse-base-multilingual-cased-v2-coreml' as const,
311-
modelSource: `${URL_PREFIX}-distiluse-base-multilingual-cased-v2/${VERSION_TAG}/coreml/distiluse_base_multilingual_cased_v2_coreml_fp32.pte`,
312-
tokenizerSource: DISTILUSE_TOKENIZER,
309+
modelSource: M.DISTILUSE_BASE_MULTILINGUAL_CASED_V2_COREML_MODEL,
310+
tokenizerSource: M.DISTILUSE_BASE_MULTILINGUAL_CASED_V2_TOKENIZER,
313311
},
314312
},
315313
};

‎packages/react-native-executorch/src/constants/modelUrls.ts‎

Lines changed: 58 additions & 43 deletions
Original file line numberDiff line numberDiff line change
@@ -573,14 +573,18 @@ export const LFM2_VL_1_6B_QUANTIZED = LFM2_5_VL_1_6B_QUANTIZED;
573573
export const LFM2_VL_450M_QUANTIZED = LFM2_5_VL_450M_QUANTIZED;
574574

575575
// Classification
576+
export const EFFICIENTNET_V2_S_XNNPACK_FP32_MODEL = `${URL_PREFIX}-efficientnet-v2-s/${VERSION_TAG}/xnnpack/efficientnet_v2_s_xnnpack_fp32.pte`;
577+
export const EFFICIENTNET_V2_S_XNNPACK_INT8_MODEL = `${URL_PREFIX}-efficientnet-v2-s/${VERSION_TAG}/xnnpack/efficientnet_v2_s_xnnpack_int8.pte`;
578+
export const EFFICIENTNET_V2_S_COREML_FP32_MODEL = `${URL_PREFIX}-efficientnet-v2-s/${VERSION_TAG}/coreml/efficientnet_v2_s_coreml_fp32.pte`;
579+
export const EFFICIENTNET_V2_S_COREML_FP16_MODEL = `${URL_PREFIX}-efficientnet-v2-s/${VERSION_TAG}/coreml/efficientnet_v2_s_coreml_fp16.pte`;
576580
const EFFICIENTNET_V2_S_MODEL =
577581
Platform.OS === `ios`
578-
? `${URL_PREFIX}-efficientnet-v2-s/${VERSION_TAG}/coreml/efficientnet_v2_s_coreml_fp32.pte`
579-
: `${URL_PREFIX}-efficientnet-v2-s/${VERSION_TAG}/xnnpack/efficientnet_v2_s_xnnpack_fp32.pte`;
582+
? EFFICIENTNET_V2_S_COREML_FP32_MODEL
583+
: EFFICIENTNET_V2_S_XNNPACK_FP32_MODEL;
580584
const EFFICIENTNET_V2_S_QUANTIZED_MODEL =
581585
Platform.OS === `ios`
582-
? `${URL_PREFIX}-efficientnet-v2-s/${VERSION_TAG}/coreml/efficientnet_v2_s_coreml_fp16.pte`
583-
: `${URL_PREFIX}-efficientnet-v2-s/${VERSION_TAG}/xnnpack/efficientnet_v2_s_xnnpack_int8.pte`;
586+
? EFFICIENTNET_V2_S_COREML_FP16_MODEL
587+
: EFFICIENTNET_V2_S_XNNPACK_INT8_MODEL;
584588

585589
/**
586590
* @category Models - Classification
@@ -599,14 +603,18 @@ export const EFFICIENTNET_V2_S_QUANTIZED = {
599603
} as const;
600604

601605
// Object detection
606+
export const SSDLITE_320_MOBILENET_V3_LARGE_XNNPACK_FP32_MODEL = `${URL_PREFIX}-ssdlite320-mobilenet-v3-large/${VERSION_TAG}/xnnpack/ssdlite320_mobilenet_v3_large_xnnpack_fp32.pte`;
607+
export const SSDLITE_320_MOBILENET_V3_LARGE_COREML_FP16_MODEL = `${URL_PREFIX}-ssdlite320-mobilenet-v3-large/${VERSION_TAG}/coreml/ssdlite320_mobilenet_v3_large_coreml_fp16.pte`;
608+
export const RF_DETR_NANO_XNNPACK_FP32_MODEL = `${URL_PREFIX}-rfdetr-nano-detector/${VERSION_TAG}/xnnpack/rfdetr_nano_xnnpack_fp32.pte`;
609+
export const RF_DETR_NANO_COREML_INT8_MODEL = `${URL_PREFIX}-rfdetr-nano-detector/${VERSION_TAG}/coreml/rfdetr_nano_coreml_int8.pte`;
602610
const SSDLITE_320_MOBILENET_V3_LARGE_MODEL =
603611
Platform.OS === 'ios'
604-
? `${URL_PREFIX}-ssdlite320-mobilenet-v3-large/${VERSION_TAG}/coreml/ssdlite320_mobilenet_v3_large_coreml_fp16.pte`
605-
: `${URL_PREFIX}-ssdlite320-mobilenet-v3-large/${VERSION_TAG}/xnnpack/ssdlite320_mobilenet_v3_large_xnnpack_fp32.pte`;
612+
? SSDLITE_320_MOBILENET_V3_LARGE_COREML_FP16_MODEL
613+
: SSDLITE_320_MOBILENET_V3_LARGE_XNNPACK_FP32_MODEL;
606614
const RF_DETR_NANO_MODEL =
607615
Platform.OS === 'ios'
608-
? `${URL_PREFIX}-rfdetr-nano-detector/${VERSION_TAG}/coreml/rfdetr_nano_coreml_int8.pte`
609-
: `${URL_PREFIX}-rfdetr-nano-detector/${VERSION_TAG}/xnnpack/rfdetr_nano_xnnpack_fp32.pte`;
616+
? RF_DETR_NANO_COREML_INT8_MODEL
617+
: RF_DETR_NANO_XNNPACK_FP32_MODEL;
610618

611619
/**
612620
* @category Models - Object Detection
@@ -683,38 +691,43 @@ export const YOLO26N_POSE = {
683691
} as const;
684692

685693
// Style transfer
694+
/**
695+
* Builds the four `(backend, precision)` URLs for a single style-transfer
696+
* style. `display` is the HF repo suffix (e.g. `rain-princess`) and `slug` is
697+
* the .pte filename token (e.g. `rain_princess`); they differ for styles
698+
* whose names contain spaces.
699+
*/
700+
export function styleTransferUrls<
701+
const Display extends string,
702+
const Slug extends string,
703+
>(display: Display, slug: Slug) {
704+
return {
705+
xnnpackBase: `${URL_PREFIX}-style-transfer-${display}/${VERSION_TAG}/xnnpack/style_transfer_${slug}_xnnpack_fp32.pte`,
706+
xnnpackQuant: `${URL_PREFIX}-style-transfer-${display}/${VERSION_TAG}/xnnpack/style_transfer_${slug}_xnnpack_int8.pte`,
707+
coremlBase: `${URL_PREFIX}-style-transfer-${display}/${VERSION_TAG}/coreml/style_transfer_${slug}_coreml_fp32.pte`,
708+
coremlQuant: `${URL_PREFIX}-style-transfer-${display}/${VERSION_TAG}/coreml/style_transfer_${slug}_coreml_fp16.pte`,
709+
};
710+
}
711+
const STYLE_TRANSFER_CANDY_URLS = styleTransferUrls('candy', 'candy');
712+
const STYLE_TRANSFER_MOSAIC_URLS = styleTransferUrls('mosaic', 'mosaic');
713+
const STYLE_TRANSFER_RAIN_PRINCESS_URLS = styleTransferUrls('rain-princess', 'rain_princess');
714+
const STYLE_TRANSFER_UDNIE_URLS = styleTransferUrls('udnie', 'udnie');
686715
const STYLE_TRANSFER_CANDY_MODEL =
687-
Platform.OS === `ios`
688-
? `${URL_PREFIX}-style-transfer-candy/${VERSION_TAG}/coreml/style_transfer_candy_coreml_fp32.pte`
689-
: `${URL_PREFIX}-style-transfer-candy/${VERSION_TAG}/xnnpack/style_transfer_candy_xnnpack_fp32.pte`;
716+
Platform.OS === `ios` ? STYLE_TRANSFER_CANDY_URLS.coremlBase : STYLE_TRANSFER_CANDY_URLS.xnnpackBase;
690717
const STYLE_TRANSFER_CANDY_QUANTIZED_MODEL =
691-
Platform.OS === `ios`
692-
? `${URL_PREFIX}-style-transfer-candy/${VERSION_TAG}/coreml/style_transfer_candy_coreml_fp16.pte`
693-
: `${URL_PREFIX}-style-transfer-candy/${VERSION_TAG}/xnnpack/style_transfer_candy_xnnpack_int8.pte`;
718+
Platform.OS === `ios` ? STYLE_TRANSFER_CANDY_URLS.coremlQuant : STYLE_TRANSFER_CANDY_URLS.xnnpackQuant;
694719
const STYLE_TRANSFER_MOSAIC_MODEL =
695-
Platform.OS === `ios`
696-
? `${URL_PREFIX}-style-transfer-mosaic/${VERSION_TAG}/coreml/style_transfer_mosaic_coreml_fp32.pte`
697-
: `${URL_PREFIX}-style-transfer-mosaic/${VERSION_TAG}/xnnpack/style_transfer_mosaic_xnnpack_fp32.pte`;
720+
Platform.OS === `ios` ? STYLE_TRANSFER_MOSAIC_URLS.coremlBase : STYLE_TRANSFER_MOSAIC_URLS.xnnpackBase;
698721
const STYLE_TRANSFER_MOSAIC_QUANTIZED_MODEL =
699-
Platform.OS === `ios`
700-
? `${URL_PREFIX}-style-transfer-mosaic/${VERSION_TAG}/coreml/style_transfer_mosaic_coreml_fp16.pte`
701-
: `${URL_PREFIX}-style-transfer-mosaic/${VERSION_TAG}/xnnpack/style_transfer_mosaic_xnnpack_int8.pte`;
722+
Platform.OS === `ios` ? STYLE_TRANSFER_MOSAIC_URLS.coremlQuant : STYLE_TRANSFER_MOSAIC_URLS.xnnpackQuant;
702723
const STYLE_TRANSFER_RAIN_PRINCESS_MODEL =
703-
Platform.OS === `ios`
704-
? `${URL_PREFIX}-style-transfer-rain-princess/${VERSION_TAG}/coreml/style_transfer_rain_princess_coreml_fp32.pte`
705-
: `${URL_PREFIX}-style-transfer-rain-princess/${VERSION_TAG}/xnnpack/style_transfer_rain_princess_xnnpack_fp32.pte`;
724+
Platform.OS === `ios` ? STYLE_TRANSFER_RAIN_PRINCESS_URLS.coremlBase : STYLE_TRANSFER_RAIN_PRINCESS_URLS.xnnpackBase;
706725
const STYLE_TRANSFER_RAIN_PRINCESS_QUANTIZED_MODEL =
707-
Platform.OS === `ios`
708-
? `${URL_PREFIX}-style-transfer-rain-princess/${VERSION_TAG}/coreml/style_transfer_rain_princess_coreml_fp16.pte`
709-
: `${URL_PREFIX}-style-transfer-rain-princess/${VERSION_TAG}/xnnpack/style_transfer_rain_princess_xnnpack_int8.pte`;
726+
Platform.OS === `ios` ? STYLE_TRANSFER_RAIN_PRINCESS_URLS.coremlQuant : STYLE_TRANSFER_RAIN_PRINCESS_URLS.xnnpackQuant;
710727
const STYLE_TRANSFER_UDNIE_MODEL =
711-
Platform.OS === `ios`
712-
? `${URL_PREFIX}-style-transfer-udnie/${VERSION_TAG}/coreml/style_transfer_udnie_coreml_fp32.pte`
713-
: `${URL_PREFIX}-style-transfer-udnie/${VERSION_TAG}/xnnpack/style_transfer_udnie_xnnpack_fp32.pte`;
728+
Platform.OS === `ios` ? STYLE_TRANSFER_UDNIE_URLS.coremlBase : STYLE_TRANSFER_UDNIE_URLS.xnnpackBase;
714729
const STYLE_TRANSFER_UDNIE_QUANTIZED_MODEL =
715-
Platform.OS === `ios`
716-
? `${URL_PREFIX}-style-transfer-udnie/${VERSION_TAG}/coreml/style_transfer_udnie_coreml_fp16.pte`
717-
: `${URL_PREFIX}-style-transfer-udnie/${VERSION_TAG}/xnnpack/style_transfer_udnie_xnnpack_int8.pte`;
730+
Platform.OS === `ios` ? STYLE_TRANSFER_UDNIE_URLS.coremlQuant : STYLE_TRANSFER_UDNIE_URLS.xnnpackQuant;
718731

719732
/**
720733
* @category Models - Style Transfer
@@ -1019,14 +1032,14 @@ export const SELFIE_SEGMENTATION = {
10191032
} as const;
10201033

10211034
// FastSAM Instance Segmentation
1035+
export const FASTSAM_S_XNNPACK_FP32_MODEL = `${URL_PREFIX}-fast-sam/${VERSION_TAG}/s/xnnpack/fast_sam_s_xnnpack_fp32.pte`;
1036+
export const FASTSAM_S_COREML_FP16_MODEL = `${URL_PREFIX}-fast-sam/${VERSION_TAG}/s/coreml/fast_sam_s_coreml_fp16.pte`;
1037+
export const FASTSAM_X_XNNPACK_FP32_MODEL = `${URL_PREFIX}-fast-sam/${VERSION_TAG}/x/xnnpack/fast_sam_x_xnnpack_fp32.pte`;
1038+
export const FASTSAM_X_COREML_FP16_MODEL = `${URL_PREFIX}-fast-sam/${VERSION_TAG}/x/coreml/fast_sam_x_coreml_fp16.pte`;
10221039
const FASTSAM_S_SEG_MODEL =
1023-
Platform.OS === 'ios'
1024-
? `${URL_PREFIX}-fast-sam/${VERSION_TAG}/s/coreml/fast_sam_s_coreml_fp16.pte`
1025-
: `${URL_PREFIX}-fast-sam/${VERSION_TAG}/s/xnnpack/fast_sam_s_xnnpack_fp32.pte`;
1040+
Platform.OS === 'ios' ? FASTSAM_S_COREML_FP16_MODEL : FASTSAM_S_XNNPACK_FP32_MODEL;
10261041
const FASTSAM_X_SEG_MODEL =
1027-
Platform.OS === 'ios'
1028-
? `${URL_PREFIX}-fast-sam/${VERSION_TAG}/x/coreml/fast_sam_x_coreml_fp16.pte`
1029-
: `${URL_PREFIX}-fast-sam/${VERSION_TAG}/x/xnnpack/fast_sam_x_xnnpack_fp32.pte`;
1042+
Platform.OS === 'ios' ? FASTSAM_X_COREML_FP16_MODEL : FASTSAM_X_XNNPACK_FP32_MODEL;
10301043

10311044
/**
10321045
* @category Models - Instance Segmentation
@@ -1052,10 +1065,12 @@ const YOLO26S_SEG_MODEL = `${URL_PREFIX}-yolo26-seg/${VERSION_TAG}/s/xnnpack/yol
10521065
const YOLO26M_SEG_MODEL = `${URL_PREFIX}-yolo26-seg/${VERSION_TAG}/m/xnnpack/yolo26_seg_m_xnnpack_fp32.pte`;
10531066
const YOLO26L_SEG_MODEL = `${URL_PREFIX}-yolo26-seg/${VERSION_TAG}/l/xnnpack/yolo26_seg_l_xnnpack_fp32.pte`;
10541067
const YOLO26X_SEG_MODEL = `${URL_PREFIX}-yolo26-seg/${VERSION_TAG}/x/xnnpack/yolo26_seg_x_xnnpack_fp32.pte`;
1068+
export const RF_DETR_NANO_SEG_XNNPACK_FP32_MODEL = `${URL_PREFIX}-rfdetr-nano-segmentation/${VERSION_TAG}/xnnpack/rfdetr_nano_xnnpack_fp32.pte`;
1069+
export const RF_DETR_NANO_SEG_COREML_INT8_MODEL = `${URL_PREFIX}-rfdetr-nano-segmentation/${VERSION_TAG}/coreml/rfdetr_nano_coreml_int8.pte`;
10551070
const RF_DETR_NANO_SEG_MODEL =
10561071
Platform.OS === 'ios'
1057-
? `${URL_PREFIX}-rfdetr-nano-segmentation/${VERSION_TAG}/coreml/rfdetr_nano_coreml_int8.pte`
1058-
: `${URL_PREFIX}-rfdetr-nano-segmentation/${VERSION_TAG}/xnnpack/rfdetr_nano_xnnpack_fp32.pte`;
1072+
? RF_DETR_NANO_SEG_COREML_INT8_MODEL
1073+
: RF_DETR_NANO_SEG_XNNPACK_FP32_MODEL;
10591074
/**
10601075
* @category Models - Instance Segmentation
10611076
*/
@@ -1133,9 +1148,9 @@ const MULTI_QA_MINILM_L6_COS_V1_MODEL = `${URL_PREFIX}-multi-qa-MiniLM-L6-cos-v1
11331148
const MULTI_QA_MINILM_L6_COS_V1_TOKENIZER = `${URL_PREFIX}-multi-qa-MiniLM-L6-cos-v1/${VERSION_TAG}/tokenizer.json`;
11341149
const MULTI_QA_MPNET_BASE_DOT_V1_MODEL = `${URL_PREFIX}-multi-qa-mpnet-base-dot-v1/${VERSION_TAG}/xnnpack/multi_qa_mpnet_base_dot_v1_xnnpack_fp32.pte`;
11351150
const MULTI_QA_MPNET_BASE_DOT_V1_TOKENIZER = `${URL_PREFIX}-multi-qa-mpnet-base-dot-v1/${VERSION_TAG}/tokenizer.json`;
1136-
const DISTILUSE_BASE_MULTILINGUAL_CASED_V2_8DA4W_MODEL = `${URL_PREFIX}-distiluse-base-multilingual-cased-v2/${VERSION_TAG}/xnnpack/distiluse_base_multilingual_cased_v2_xnnpack_8da4w.pte`;
1137-
const DISTILUSE_BASE_MULTILINGUAL_CASED_V2_COREML_MODEL = `${URL_PREFIX}-distiluse-base-multilingual-cased-v2/${VERSION_TAG}/coreml/distiluse_base_multilingual_cased_v2_coreml_fp32.pte`;
1138-
const DISTILUSE_BASE_MULTILINGUAL_CASED_V2_TOKENIZER = `${URL_PREFIX}-distiluse-base-multilingual-cased-v2/${VERSION_TAG}/tokenizer.json`;
1151+
export const DISTILUSE_BASE_MULTILINGUAL_CASED_V2_8DA4W_MODEL = `${URL_PREFIX}-distiluse-base-multilingual-cased-v2/${VERSION_TAG}/xnnpack/distiluse_base_multilingual_cased_v2_xnnpack_8da4w.pte`;
1152+
export const DISTILUSE_BASE_MULTILINGUAL_CASED_V2_COREML_MODEL = `${URL_PREFIX}-distiluse-base-multilingual-cased-v2/${VERSION_TAG}/coreml/distiluse_base_multilingual_cased_v2_coreml_fp32.pte`;
1153+
export const DISTILUSE_BASE_MULTILINGUAL_CASED_V2_TOKENIZER = `${URL_PREFIX}-distiluse-base-multilingual-cased-v2/${VERSION_TAG}/tokenizer.json`;
11391154
const PARAPHRASE_MULTILINGUAL_MINILM_L12_V2_QUANTIZED_MODEL = `${URL_PREFIX}-paraphrase-multilingual-MiniLM-L12-v2/${VERSION_TAG}/xnnpack/paraphrase_multilingual_minilm_l12_v2_xnnpack_8da4w.pte`;
11401155
const PARAPHRASE_MULTILINGUAL_MINILM_L12_V2_TOKENIZER = `${URL_PREFIX}-paraphrase-multilingual-MiniLM-L12-v2/${VERSION_TAG}/tokenizer.json`;
11411156
const CLIP_VIT_BASE_PATCH32_TEXT_MODEL = `${URL_PREFIX}-clip-vit-base-patch32/${VERSION_TAG}/xnnpack/clip_vit_base_patch32_text_xnnpack_fp32.pte`;

0 commit comments

Comments
 (0)