@@ -385,13 +385,15 @@ const EFFICIENTNET_V2_S_QUANTIZED_MODEL =
385385 * @category Models - Classification
386386 */
387387export const EFFICIENTNET_V2_S = {
388+ modelName : 'efficientnet-v2-s' ,
388389 modelSource : EFFICIENTNET_V2_S_MODEL ,
389390} ;
390391
391392/**
392393 * @category Models - Classification
393394 */
394395export 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 */
458460export const STYLE_TRANSFER_CANDY = {
461+ modelName : 'style-transfer-candy' ,
459462 modelSource : STYLE_TRANSFER_CANDY_MODEL ,
460463} ;
461464
462465/**
463466 * @category Models - Style Transfer
464467 */
465468export 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 */
472476export const STYLE_TRANSFER_MOSAIC = {
477+ modelName : 'style-transfer-mosaic' ,
473478 modelSource : STYLE_TRANSFER_MOSAIC_MODEL ,
474479} ;
475480
476481/**
477482 * @category Models - Style Transfer
478483 */
479484export 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 */
486492export 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 */
493500export 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 */
500508export const STYLE_TRANSFER_UDNIE = {
509+ modelName : 'style-transfer-udnie' ,
501510 modelSource : STYLE_TRANSFER_UDNIE_MODEL ,
502511} ;
503512
504513/**
505514 * @category Models - Style Transfer
506515 */
507516export 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 */
735745export 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 */
742753export 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 */
793805export 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