A Collection of Google's On-Device Models
Help us complete the list
- To contribute go to C:\Users\YOUR_PC_USER\AppData\Local\Google\Chrome\User Data\optimization_guide_model_store
- If you find a new non-empty folder not listed here please upload it to this repo
List of All Available Models
Following is the complete list of machine learning models in Chrome many of which are on your device. They are located in your User Data folder and you can easily check to see which ones you have as they are all in numbered folders.
Mapping of folder names to optimization target descriptions
OPTIMIZATION_TARGETS = {
"0": "UNKNOWN",
"1": "PAINFUL_PAGE_LOAD",
"2": "LANGUAGE_DETECTION",
"3": "PAGE_TOPICS",
"4": "SEGMENTATION_NEW_TAB",
"5": "SEGMENTATION_SHARE",
"6": "SEGMENTATION_VOICE",
"7": "MODEL_VALIDATION",
"8": "PAGE_ENTITIES",
"9": "NOTIFICATION_PERMISSION_PREDICTIONS",
"10": "SEGMENTATION_DUMMY",
"11": "SEGMENTATION_CHROME_START_ANDROID",
"12": "SEGMENTATION_QUERY_TILES",
"13": "PAGE_VISIBILITY",
"15": "PAGE_TOPICS_V2",
"16": "SEGMENTATION_CHROME_LOW_USER_ENGAGEMENT",
"17": "SEGMENTATION_FEED_USER",
"18": "CONTEXTUAL_PAGE_ACTION_PRICE_TRACKING",
"19": "TEXT_CLASSIFIER",
"20": "GEOLOCATION_PERMISSION_PREDICTIONS",
"21": "SEGMENTATION_SHOPPING_USER",
"22": "SEGMENTATION_CHROME_START_ANDROID_V2",
"23": "SEGMENTATION_SEARCH_USER",
"24": "OMNIBOX_ON_DEVICE_TAIL_SUGGEST",
"25": "CLIENT_SIDE_PHISHING",
"26": "OMNIBOX_URL_SCORING",
"27": "SEGMENTATION_DEVICE_SWITCHER",
"28": "SEGMENTATION_ADAPTIVE_TOOLBAR",
"29": "SEGMENTATION_TABLET_PRODUCTIVITY_USER",
"30": "CLIENT_SIDE_PHISHING_IMAGE_EMBEDDER",
"31": "NEW_TAB_PAGE_HISTORY_CLUSTERS_MODULE_RANKING",
"32": "WEB_APP_INSTALLATION_PROMO",
"33": "TEXT_EMBEDDER",
"34": "VISUAL_SEARCH_CLASSIFICATION",
"35": "SEGMENTATION_BOTTOM_TOOLBAR",
"36": "AUTOFILL_FIELD_CLASSIFICATION",
"37": "SEGMENTATION_IOS_MODULE_RANKER",
"38": "SEGMENTATION_DESKTOP_NTP_MODULE",
"39": "PRELOADING_HEURISTICS",
"40": "TEXT_SAFETY",
"41": "SEGMENTATION_ANDROID_HOME_MODULE_RANKER",
"42": "COMPOSE",
"43": "PASSAGE_EMBEDDER",
"44": "PHRASE_SEGMENTATION",
"45": "SEGMENTATION_COMPOSE_PROMOTION",
"46": "URL_VISIT_RESUMPTION_RANKER",
"47": "CAMERA_BACKGROUND_SEGMENTATION",
"48": "MODEL_EXECUTION_FEATURE_HISTORY_SEARCH",
"49": "MODEL_EXECUTION_FEATURE_PROMPT_API",
"50": "SEGMENTATION_METRICS_CLUSTERING",
"51": "MODEL_EXECUTION_FEATURE_SUMMARIZE",
"52": "PASSWORD_MANAGER_FORM_CLASSIFICATION",
"53": "NOTIFICATION_CONTENT_DETECTION",
"54": "MODEL_EXECUTION_FEATURE_HISTORY_QUERY_INTENT",
"55": "MODEL_EXECUTION_FEATURE_SCAM_DETECTION",
"56": "MODEL_EXECUTION_FEATURE_PERMISSIONS_AI",
"57": "EXPERIMENTAL_EMBEDDER",
"58": "SEGMENTATION_FEDCM_USER",
"59": "MODEL_EXECUTION_FEATURE_WRITING_ASSISTANCE_API"
}
Source: DEJAN
Inference Providers
NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API:
The model has no library tag.