optimum-neuron-cache / registry.json
kctung's picture
Add google/vit-base-patch16-224-in21k in registry for NeuronHash e1e014bd0bfa0712d7cd564b786b54ac2fdde6321e8b96765ce9e84e4d300aa5895d9fe4f464c5424896bebc03163337835b583b814860ca4adb3095a12b9482
25f7d04
raw
history blame
15.8 kB
{"2.6.0.19+3d819e565": {"Helsinki-NLP/opus-mt-en-ro": {"model_name_or_path": "Helsinki-NLP/opus-mt-en-ro", "model_hash": "c6eb4873140e01f6383179ebd4d59197a075e40673d2250e7d94c06f9ec6bd349c05f80931761df64ba56c7db30b3422c9134ce8b99d14bc246def050b190d49", "features": [{"input_shapes": [["input_ids", [4, 512]], ["attention_mask", [4, 512]], ["labels", [4, 512]]], "precision": "torch.float32", "num_neuron_cores": 1, "neuron_hash": "718bcffb8118309b247a5405b15c1596dc982a85fda1278138967f181b9061f4bc98ecbc6e92f0417ff748612b929ebff05442b7640c29293d629b5dbcbf2096"}, {"input_shapes": [["input_ids", [16, 512]], ["attention_mask", [16, 512]], ["labels", [16, 512]]], "precision": "torch.float32", "num_neuron_cores": 1, "neuron_hash": "cb51842fd8f531973d8c4a3bbc68d4ab0745abecb799bdcf4a6246b56892a97e9d75b7b7e744769b0ff5493a760992beeb4d2f9972aefe680b1eb2a00b70f93f"}, {"input_shapes": [["input_ids", [2, 512]], ["attention_mask", [2, 512]], ["labels", [2, 512]]], "precision": "torch.float32", "num_neuron_cores": 1, "neuron_hash": "4e3254e4f824724d529aa8d7aafd0624f762f7efabaaf06b249e4d9181109285f5beabfaa02c42d5c924659e069467af7ea35227ea31995b5370d5ab6ac7ef2b"}]}}, "2.5.0.28+1be23f232": {"EleutherAI/gpt-neo-1.3B": {"model_name_or_path": "EleutherAI/gpt-neo-1.3B", "model_hash": "6de0d74fe435f83574538bed7a89d15755ea550f176bb9cb3281e3991a8cbc61e4ebee39410c6bca6f4fa58b0a7ce205bd90516c7f2fde3c6bff5b8abd85a9a4", "features": [{"input_shapes": [], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "b3267ae27a9e66845ba02c22cca97d4c1f430c5fcdd2a813e521fb940bb21bc8d275decc291290c0641b5c529235a37a4b206f5cb6c3c62ab00d70de71d8bec0"}]}, "hf-internal-testing/tiny-random-gpt_neo": {"model_name_or_path": "hf-internal-testing/tiny-random-gpt_neo", "model_hash": "9978a18b39b95c6a2540313c7573d6dc32981e5d14dfc62c008fedb47d34f156a8b58c4dca195a038107cceec04963b13a7872597c0d5b869c6f9a9adafe5980", "features": [{"input_shapes": [["attention_mask", [2, 512]], ["labels", [2, 512]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "db4f64ac68b72a633c95bde0953378bcdacf624da48848c4a0c9531029e62b4ca679b1f16b79f514cfe3ec419061e537442baab16285bf67c9a6c998f840d77c"}, {"input_shapes": [["attention_mask", [2, 512]], ["labels", [2, 512]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "3911d4f03f9cb7bc510f4089b97a40d7a3e05d67bf447bd7e61d6256c775ec5387f8f4a932db4fe21993f31ed7a083c11c373c4f42dbdae431980046ea2570b6"}, {"input_shapes": [["attention_mask", [2, 512]], ["labels", [2, 512]]], "precision": "torch.float32", "num_neuron_cores": 1, "neuron_hash": "f4bbc19059a6284566295699ea87b303438414abbb759a0a15f724c7bdcf0cfb6dd0d063e26163df77c6b7539c1e0f298c292df87d8ad96f14e33ed046dce227"}, {"input_shapes": [["attention_mask", [2, 128]], ["labels", [2, 128]]], "precision": "torch.float32", "num_neuron_cores": 1, "neuron_hash": "6b09ba88e4bc3f63d55273e8ceefaddb2c54f9d63c2b1b4ee0e24e0c92cbb92fb1e45f41832723c680f42ee202b7f232235531d13ae8ac547531c38b07517c97"}, {"input_shapes": [], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "20120d8facc5b15d25f1a9c2d06f9aaaca16485d080bfe3549f9fae3fab129c9611b21e25144be93d667750c2ce544e14794bc7390776ac93f105f5795d9cbd0"}]}}, "2.7.0.40+f7c6cf2a3": {"bert-base-uncased": {"model_name_or_path": "bert-base-uncased", "model_hash": "9c3673539c14b1b97107640f18ca804ee69a1575250db52ebabfeda4a41007db3a4ce21f46b934e4d311184105d34107801273277d4b601ef4eb687cc236692a", "features": [{"input_shapes": [["labels", [16]], ["token_type_ids", [16, 128]], ["attention_mask", [16, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "5129a8c980da55dee63dbd548567fbf10c049c16d1b10cd7e687dbd29889bb6c5e3256a4b18451236af2f4ac2e1cff584301500f9b9a58feb92ebe3131fa623d"}, {"input_shapes": [["labels", [8]], ["token_type_ids", [8, 128]], ["attention_mask", [8, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "a68fa2104fd72a21b11563f84da03bdf5c5d8c65298fe91d9585341b1d5ead3d6362fc8e50ce59ddaab6d2ed4422f63f38c3ade204727576643c5924c5723cda"}]}, "huggyllama/llama-7b": {"model_name_or_path": "huggyllama/llama-7b", "model_hash": "1297b5fd2d992634a5e5ba975dbb15ecab18d6ff60e8cadb45f92a5d18b0e6d2fdb033793022aa4e2e6b639005c96c579edf3f8d3858d4b5cda11dcb3546e59f", "features": [{"input_shapes": [["attention_mask", [1, 1024]], ["labels", [1, 1024]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "fadbfdaaeb661bdbbc888b031a38fb87995d53c68c4c31fa9f99a584ff288e473cf76ceaabd374a2b46bd3bc48735785ad98468fc536c8d44fbcb0dec7b181fa"}]}, "patrickvonplaten/t5-tiny-random": {"model_name_or_path": "patrickvonplaten/t5-tiny-random", "model_hash": "b5d927c9f7066674483250e96aa3168c7f898daa063e56e4f2e2c3ae6a2b37dfae19c36bf61edf8d8c5aac6d9615bacaef3813b96ef4720b3bda73335bd7c436", "features": [{"input_shapes": [["attention_mask", [4, 1024]], ["labels", [4, 128]], ["decoder_input_ids", [4, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "6e82f4183d2a55622c22fc3605db4e4b7d70dc6f7049562a7cfd5474358770d9aceb329838418e9789c9ac12cf7a31efa04b65b9070fbedcda008ef54e7a797a"}]}}, "2.8.0.25+a3ad0f342": {"prajjwal1/bert-tiny": {"model_name_or_path": "prajjwal1/bert-tiny", "model_hash": "8579f3f2e02f4f96103b78b4e69520edabdbc30b4bfb96b01189b8b859e82596aecbaa858c263588e52dcee3b1395f6a767f4d1bdc94ac283888fe9460635cc3", "features": [{"input_shapes": [["labels", [128]], ["token_type_ids", [128, 128]], ["attention_mask", [128, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "578b3c544f3df77ff6f9a15342a2edb85638d68a7528278ddc6c51dc4bcae528409d7a3f799f6934a87934d0489cd8d6c893b2ab33c9e7da16ddd5442af917c3"}, {"input_shapes": [["token_type_ids", [8, 128]], ["attention_mask", [8, 128]], ["labels", [8, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "e5497a57bd0a8b3ec8f1cae50d5ee05c5e49f50ebed71ec029d2b80334465c89513e2f5fe2c300e1eecb09c5cd2e6b2c2e07ff60b045e5b4d3b76d1abd37ccfa"}, {"input_shapes": [["token_type_ids", [8, 128]], ["attention_mask", [8, 128]], ["labels", [8, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "e5ba6d5cd495e0c8f0e4562e8f109e0c060a89d14e9c1e142b3a65fa88d77cec2199cd03dda1350c0cceb8d192577b267bcfc1012dc6ab6d3b1e181268266204"}, {"input_shapes": [["token_type_ids", [8, 128]], ["attention_mask", [8, 128]], ["labels", [8, 128]]], "precision": "torch.float32", "num_neuron_cores": 1, "neuron_hash": "843530c5773322631f8d3265f00d5f4c69fb941814cdc73781d30043b71dca443dcb12e0078ac452580e15aa32cc0868141b1afb1f6ca0e65849e1e52ccb33f3"}, {"input_shapes": [["token_type_ids", [8, 200]], ["attention_mask", [8, 200]], ["labels", [8, 200]]], "precision": "torch.float32", "num_neuron_cores": 1, "neuron_hash": "1220fed9fc14d3bffeb9e15a738eba2d568cfc71cd66be27ce657427f6dd6df2de6871334ea179f97668aa79408b657bf31872c49ab4fc56cbe2a580243d6a91"}, {"input_shapes": [["token_type_ids", [8, 128]], ["attention_mask", [8, 128]], ["labels", [8, 128]]], "precision": "torch.float32", "num_neuron_cores": 1, "neuron_hash": "2f438c94c2cbc95ffcdca2ffcdcf7ac75983c1fc8d3358635726191a0d2b3b1a2b2aae08b5b9da1dfa28954336aacd72b467ce802ae12b71ea35ecd600d12aba"}, {"input_shapes": [["token_type_ids", [8, 202]], ["attention_mask", [8, 202]], ["labels", [8, 202]]], "precision": "torch.float32", "num_neuron_cores": 1, "neuron_hash": "e427b08bd24ce6352777d35cdfa7cf1ee4b95436d7dc71e3d0e4a1a8190510112a0dd29912c19b720961b69ac384d3ed31ad2c79e726cd6d56817f4ff6a2ac76"}, {"input_shapes": [["token_type_ids", [8, 202]], ["attention_mask", [8, 202]], ["labels", [8, 202]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "d0c2d506e6d0415291cbad40fb28aeef2729ff5507a1e7c99af5247f82eed39e899ff328663ff1a0e6c5238048359fd379b824cc331c41ec3dde38400ffa10ea"}, {"input_shapes": [["token_type_ids", [5, 202]], ["attention_mask", [5, 202]], ["labels", [5, 202]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "35d7cd6da040d3257825dace63945d8fd3d443e5bc65b98b1818993eb103142f949af10f0fa22d9fc83d91e99a04db787ab1b077c2575dc15b9d4b59117a617b"}, {"input_shapes": [["token_type_ids", [8, 202]], ["attention_mask", [8, 202]], ["labels", [8, 202]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "cee9e429e4c5a8f13f57b13bb9b5be700e36af77344a284920909db6d09e1752051508cef34e750c0c17a03e1a4634701b75def196c93ab3b8148391a00d5edd"}, {"input_shapes": [["labels", [128]], ["token_type_ids", [128, 128]], ["attention_mask", [128, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "221671508bee115285054e3b6cfa9b6538ed258c4c3fb36da2fb82f1287a0c197d8e7a418bc67da317238ec441ddca90fdaa11ec9afe426ae5dc876b451d4f53"}, {"input_shapes": [["labels", [21]], ["token_type_ids", [21, 128]], ["attention_mask", [21, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "c9d8029962fefead3547102baac980a9ae3d9215bcc9281b1c6c96a82f7d8b8ac36c0d7b1f808e424a2e034c824b5362246b654a4f708e1d5479adaefbfc656c"}, {"input_shapes": [["labels", [21]], ["token_type_ids", [21, 128]], ["attention_mask", [21, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "b3adb557bd625768c9bc9750d7d1370b3fef7bd2a3a807c300873a23bed6facbc3431b0dc4878877682cac9ce9ffd465a66be2c227c2dc085a37160d373b1164"}, {"input_shapes": [["labels", [128]], ["token_type_ids", [128, 128]], ["attention_mask", [128, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "459eeaa4c7f405f8f6548c3ab8ca915920e4d8f690047acc23a81fd5049524b8ae4b2d85dc989d59986b70090d0a49cf7b98560cdc24143a084fceb2f9bf1ab1"}, {"input_shapes": [["labels", [21]], ["token_type_ids", [21, 128]], ["attention_mask", [21, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "a61837e1034f1e3db9a96c312317db965534a89bf955446430742dac90b9520199743fe6907a885855e483705aa181089d3203ebee005f6becc7b115b6df11a3"}, {"input_shapes": [["labels", [128]], ["token_type_ids", [128, 128]], ["attention_mask", [128, 128]]], "precision": "torch.float32", "num_neuron_cores": 1, "neuron_hash": "f151833e2548e643246ae8bb6d64fba290b3616b3dc5135b355052604c0157679a1ce186b5b7bcde2ad5623fd87f8cbb49f7adfe6eabbf48e98b34bf0c883221"}, {"input_shapes": [["labels", [21]], ["token_type_ids", [21, 128]], ["attention_mask", [21, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "d1bd99994aa8eebe86c8a6027ec4c9d5b432fb5e5f26aa1c3802d3709ce68f556e119d937f549a196bf983d097c868bf9fb14c045860648b3c7b57a30157e00c"}, {"input_shapes": [["labels", [21]], ["token_type_ids", [21, 128]], ["attention_mask", [21, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "6eb6134ecce59699d0e39a79ff65592020646ee92a5c2ab30523669d4ebfeed248760dec3b412cf5aa66655a43d008814971505ef800a26c4c5d426702153737"}, {"input_shapes": [["labels", [21]], ["token_type_ids", [21, 128]], ["attention_mask", [21, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "ee4f98745758d3935b9bfe71f0981e594791914d0bebb28d75113d0a4abb4f8efea4f95bd855ed74cee812a69d5c04bb0c1769939b53b9e4b8c5aff2d1c6f968"}, {"input_shapes": [["labels", [21]], ["token_type_ids", [21, 128]], ["attention_mask", [21, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "6a65613d85460c9a0539c0cb59794843519c069f9594e0d01649af67fdd6b7a22e42527319d00ed6e3e49f83c7781182e410d4a96b278181d6c9e28273b0c35e"}, {"input_shapes": [["labels", [128]], ["token_type_ids", [128, 128]], ["attention_mask", [128, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "3e33a65cb9a0c9e408b0757261edb020c67d2fd86aa0423a4d42b077b6bd63118ccc5c107e14498013a98de615c37be81056c11ce52f5b7499fd50f81166cf5c"}]}, "arampacha/roberta-tiny": {"model_name_or_path": "arampacha/roberta-tiny", "model_hash": "e7ee48a33fad3c376e92593e380938b09723f96016c45e930628e877681aa12f0548822bdb635e9d184cf48c1f4eb8b79696d20d2a84897cc970730b5259e1a9", "features": [{"input_shapes": [["attention_mask", [8, 128]], ["labels", [8, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "e7d1138eefd203b89d0898b720c699890857760f0274de68cf8b410fef43e4f4626dcba3a33423503e36b1c36be5ccf83811601f28733245e580aa9f3fdd9c6c"}, {"input_shapes": [["attention_mask", [2, 128]], ["labels", [2, 128]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "0d67b0a99b31c0b59c475991909350148b3f659915875884efc3612598f49c5afbca94e17c53de14fd827fb021f0036c4171f80683555cce6fb0d78465c40a3c"}]}, "google/vit-base-patch16-224-in21k": {"model_name_or_path": "google/vit-base-patch16-224-in21k", "model_hash": "ff1961ec42b32516dd1cd7de9d0d9244078512ffb4d92850d1df36b7c07cdd68e9296d6ad4a54c395ad2fe2ee377280f20faafcf00269da69ead6a436101b6ef", "features": [{"input_shapes": [["labels", [8]]], "precision": "torch.float32", "num_neuron_cores": 1, "neuron_hash": "869b22ff2e6162e366b5774f0f1f3c13e2cec03de4416c968acf9d72e2279c8a488021736956420e80bdee221b2c6ccfb5fbcb11d8e376f3f8228c33b0d2d9e3"}, {"input_shapes": [["labels", [2]]], "precision": "torch.float32", "num_neuron_cores": 1, "neuron_hash": "04e78d1e657adc2268d7bff45caa5c686d68367195f720722a080404c11e73f806a82f8d5afcbeef3bc59b3ac5c68df518c8aabd41cff39742726931ceca4a72"}]}, "google/flan-t5-small": {"model_name_or_path": "google/flan-t5-small", "model_hash": "febc9574dc7fbfb9057d75d8f9e8363b85dd39bece42e0b2f21e3399cf5b197cded1b2d7461593207d89bc5d45240371480f03ff1c68ff289e31ca6123522be6", "features": [{"input_shapes": [["attention_mask", [1, 1024]], ["labels", [1, 128]], ["decoder_input_ids", [1, 128]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "387e414d84d7074fb686211a5bdf10847e8d35d17e19141104c9cc04e829eeb010e3accc77f1ce962b80053897d696b1ece90ef0b947695b4196d10db1c3264e"}, {"input_shapes": [["attention_mask", [2, 1024]], ["labels", [2, 128]], ["decoder_input_ids", [2, 128]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "38846280f25d22e0a80875876f726cc2d27c011a7ea2bc4d692378aacf9101a25be645f02f4a49b9bbf12066c743afc22937905fbfcca37a09015667c1b87a26"}, {"input_shapes": [["attention_mask", [4, 1024]], ["labels", [4, 128]], ["decoder_input_ids", [4, 128]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "cf9b9dcce85432f2fcfb9057a48bfe972923aef9cd9e88d917594969cce665447ae008cbca34f4dc5b3a3bf4257076dd61e0071b103ffcde79e207f65b86ab20"}]}}, "2.11.0.18+fada6114a": {"google/vit-base-patch16-224-in21k": {"model_name_or_path": "google/vit-base-patch16-224-in21k", "model_hash": "ea88be553a835f83559b7c494d4b00971f50f7a13e3aec51ab30386fcf74223c8b7b0ca80faf3b9b844387e7525f1ac570327ba82be2cb5e96ed4068bf9ba637", "features": [{"input_shapes": [["pixel_values", [16, 3, 224, 224]], ["labels", [16]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "62d3162b6f000625bc02c2edaa0adab39d15d2309de8505b4b1cdeeaa655b42d953772e355da099f041ee55523420a52912244c3633372466dfbab0a38ba774a"}, {"input_shapes": [["pixel_values", [6, 3, 224, 224]], ["labels", [6]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "e1e014bd0bfa0712d7cd564b786b54ac2fdde6321e8b96765ce9e84e4d300aa5895d9fe4f464c5424896bebc03163337835b583b814860ca4adb3095a12b9482"}]}}, "2.11.0.6+71b2938aa": {"google/vit-base-patch16-224-in21k": {"model_name_or_path": "google/vit-base-patch16-224-in21k", "model_hash": "ea88be553a835f83559b7c494d4b00971f50f7a13e3aec51ab30386fcf74223c8b7b0ca80faf3b9b844387e7525f1ac570327ba82be2cb5e96ed4068bf9ba637", "features": [{"input_shapes": [["pixel_values", [16, 3, 224, 224]], ["labels", [16]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "2db27bc94dc6bf8d4379677bc453bd53f8cdfc6e19244f94149ad17da04c536e07e62743ea9220eb6832265bb9265015a67d153a0f1c733c9a4d17467ed31bf7"}]}}}