Skip to content

Commit 8384e26

Browse files
committed
feat: add modelName property for changed models
1 parent 616b0be commit 8384e26

1 file changed

Lines changed: 13 additions & 0 deletions

File tree

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

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -385,13 +385,15 @@ const EFFICIENTNET_V2_S_QUANTIZED_MODEL =
385385
* @category Models - Classification
386386
*/
387387
export const EFFICIENTNET_V2_S = {
388+
modelName: 'efficientnet-v2-s',
388389
modelSource: EFFICIENTNET_V2_S_MODEL,
389390
};
390391

391392
/**
392393
* @category Models - Classification
393394
*/
394395
export const EFFICIENTNET_V2_S_QUANTIZED = {
396+
modelName: 'efficientnet-v2-s-quantized',
395397
modelSource: EFFICIENTNET_V2_S_QUANTIZED_MODEL,
396398
};
397399

@@ -456,55 +458,63 @@ const STYLE_TRANSFER_UDNIE_QUANTIZED_MODEL =
456458
* @category Models - Style Transfer
457459
*/
458460
export const STYLE_TRANSFER_CANDY = {
461+
modelName: 'style-transfer-candy',
459462
modelSource: STYLE_TRANSFER_CANDY_MODEL,
460463
};
461464

462465
/**
463466
* @category Models - Style Transfer
464467
*/
465468
export const STYLE_TRANSFER_CANDY_QUANTIZED = {
469+
modelName: 'style-transfer-candy-quantized',
466470
modelSource: STYLE_TRANSFER_CANDY_QUANTIZED_MODEL,
467471
};
468472

469473
/**
470474
* @category Models - Style Transfer
471475
*/
472476
export const STYLE_TRANSFER_MOSAIC = {
477+
modelName: 'style-transfer-mosaic',
473478
modelSource: STYLE_TRANSFER_MOSAIC_MODEL,
474479
};
475480

476481
/**
477482
* @category Models - Style Transfer
478483
*/
479484
export const STYLE_TRANSFER_MOSAIC_QUANTIZED = {
485+
modelName: 'style-transfer-mosaic-quantized',
480486
modelSource: STYLE_TRANSFER_MOSAIC_QUANTIZED_MODEL,
481487
};
482488

483489
/**
484490
* @category Models - Style Transfer
485491
*/
486492
export const STYLE_TRANSFER_RAIN_PRINCESS = {
493+
modelName: 'style-transfer-rain-princess',
487494
modelSource: STYLE_TRANSFER_RAIN_PRINCESS_MODEL,
488495
};
489496

490497
/**
491498
* @category Models - Style Transfer
492499
*/
493500
export const STYLE_TRANSFER_RAIN_PRINCESS_QUANTIZED = {
501+
modelName: 'style-transfer-rain-princess-quantized',
494502
modelSource: STYLE_TRANSFER_RAIN_PRINCESS_QUANTIZED_MODEL,
495503
};
496504

497505
/**
498506
* @category Models - Style Transfer
499507
*/
500508
export const STYLE_TRANSFER_UDNIE = {
509+
modelName: 'style-transfer-udnie',
501510
modelSource: STYLE_TRANSFER_UDNIE_MODEL,
502511
};
503512

504513
/**
505514
* @category Models - Style Transfer
506515
*/
507516
export const STYLE_TRANSFER_UDNIE_QUANTIZED = {
517+
modelName: 'style-transfer-udnie-quantized',
508518
modelSource: STYLE_TRANSFER_UDNIE_QUANTIZED_MODEL,
509519
};
510520

@@ -733,13 +743,15 @@ const CLIP_VIT_BASE_PATCH32_IMAGE_QUANTIZED_MODEL = `${URL_PREFIX}-clip-vit-base
733743
* @category Models - Image Embeddings
734744
*/
735745
export const CLIP_VIT_BASE_PATCH32_IMAGE = {
746+
modelName: 'clip-vit-base-patch32-image',
736747
modelSource: CLIP_VIT_BASE_PATCH32_IMAGE_MODEL,
737748
};
738749

739750
/**
740751
* @category Models - Image Embeddings
741752
*/
742753
export const CLIP_VIT_BASE_PATCH32_IMAGE_QUANTIZED = {
754+
modelName: 'clip-vit-base-patch32-image-quantized',
743755
modelSource: CLIP_VIT_BASE_PATCH32_IMAGE_QUANTIZED_MODEL,
744756
};
745757

@@ -791,6 +803,7 @@ export const MULTI_QA_MPNET_BASE_DOT_V1 = {
791803
* @category Models - Text Embeddings
792804
*/
793805
export const CLIP_VIT_BASE_PATCH32_TEXT = {
806+
modelName: 'clip-vit-base-patch32-text',
794807
modelSource: CLIP_VIT_BASE_PATCH32_TEXT_MODEL,
795808
tokenizerSource: CLIP_VIT_BASE_PATCH32_TEXT_TOKENIZER,
796809
};

0 commit comments

Comments
 (0)