optimum-neuron-cache / registry.json
juliensimon's picture
Add TinyLlama/TinyLlama-1.1B-Chat-v0.6 in registry for NeuronHash a323cfa789ff7ee78a7930ecabb43afd182dd97e27ec302d2747e36c34fc5d6f4e5a455a69b0b133ce38fdfc97e3456e3b0f748dcbf602981a8ad96304e4cad9
81fae57 verified
{"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"}, {"input_shapes": [["pixel_values", [6, 3, 224, 224]], ["labels", [6]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "6b1caeb507bf506bf108642da83ecbaa6715932a2e8f45c2c066c318219afd8b1f2037faf9a9b40263950fdf27ebdf2314f8a49063ffcf6aaa3e1d1ee57b86d1"}]}}, "2.10.0.35+3817a0c8c": {"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": "177c533efe1d0a9d6bd0554a50f740a7b37df426354681f377495acbb3a2de9c03aad9a365b7d7b11f7bfed74d3df9a6f486791340d73bd1b24c9b729bd22804"}, {"input_shapes": [["pixel_values", [6, 3, 224, 224]], ["labels", [6]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "28bb562e3b7654f8ee752d321477b267854956ec9c855fea31254ce7f18a889261cdbcd59c507a87f2e9fe590b9010beae7633ca7753c01545aaa867c829d4fc"}, {"input_shapes": [["pixel_values", [6, 3, 224, 224]], ["labels", [6]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "b09a499c6ffad4ee1f7cbc2aecc574e4b4cfb89afbfa2ef5ba5dd92033a301b5baab63a584a63b7f58206b8e0bf853f8065da68f4c0b76feef8db64db61ea035"}, {"input_shapes": [["pixel_values", [16, 3, 224, 224]], ["labels", [16]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "5b0f7a92406554f50b7abd50d8f7b9ed11ac5f0586a218a8d7e5d8233a447e594d3e74ed35b542ca43ed995a5c678a6a0fffbfbd3cc04cf1b6f4eb07e02cd35b"}, {"input_shapes": [["pixel_values", [16, 3, 224, 224]], ["labels", [16]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "fbd1feef89af43db00b114ae6cccc88cd912a5225646f4d1ea7e93ed81ca8e606f7fa0f3a58694da328641dc1009a4d33c750f939777fec4d40681b10216ef4c"}, {"input_shapes": [["pixel_values", [6, 3, 224, 224]], ["labels", [6]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "24440c9e531d4a10d820c9eba0a1b2b601de022770c331e6a873df1f2561d316abc1f4aeb15025b728d25ea9638eddf86cac62161a4a63f557eb9346e3df2ca8"}, {"input_shapes": [["pixel_values", [32, 3, 224, 224]], ["labels", [32]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "55da2f50afeb32b5a2f24f39e3d2ff79a8d8e7313247ca3aaee4c167f81c4c1692e1f8d31db657b7c72558644548320731b78f28fd65effb3d50e4830efe7e79"}, {"input_shapes": [["pixel_values", [2, 3, 224, 224]], ["labels", [2]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "2db025f9820efbdc009036e0994c7267a01cf8fc169923c4836437544af8f5894ba8f58b0b13d519f8fa36fcb5e8d663ce83eed1b7c1bf89b1fcd06b8dc1f0b3"}, {"input_shapes": [["pixel_values", [16, 3, 224, 224]], ["labels", [16]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "04d755838ee93c820ec3a7ea9be2c4b4c37dedf36278e4d3f89e0418936953cb1867dc3499f0df645a14f755ec8ec69156a8088de517249c1e04bf68a75de403"}, {"input_shapes": [["pixel_values", [6, 3, 224, 224]], ["labels", [6]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "87400e737bae35747a91790622536baf7e48bfc8ac44da05c9dfe03b373c552b8740a037b64e2417cd307eaa8be5586dcae8f6be0b61c0f1cd0d22681379cc3f"}]}}, "2.11.0.34+c5231f848": {"hf-internal-testing/tiny-random-bert": {"model_name_or_path": "hf-internal-testing/tiny-random-bert", "model_hash": "446f651a6b0781e1fda1466641d1821a0e0e50aa9a4a3a1040faa091b4d33a5887682ddf7c5efe42c4e78508c9ab98a049f38195971d840cdacff15adea8c7f0", "features": [{"input_shapes": [["labels", [2]], ["input_ids", [2, 128]], ["token_type_ids", [2, 128]], ["attention_mask", [2, 128]]], "precision": "torch.float32", "num_neuron_cores": 8, "neuron_hash": "28b11e90c81fc3346a3d19636697636563f517cec016ff9263897e936436ae73b1d3ca035f7a3cc4fdb83299fb895112e8086da2b24851b4a57424430e61a331"}, {"input_shapes": [["labels", [2]], ["input_ids", [2, 128]], ["token_type_ids", [2, 128]], ["attention_mask", [2, 128]]], "precision": "torch.float32", "num_neuron_cores": 8, "neuron_hash": "e76c700be6f003136000c68c459b125943f01f17a61ae0ded46ac0089dfc99bd238d9499afbc5c5a7b0c2b86ec77d82b1c5f0ac235c1e6dcf92867343e7dabbd"}]}, "t5-small": {"model_name_or_path": "t5-small", "model_hash": "655ab7e174e6b7581d58b6e9a29a2213af2176fe5a709e6f794677aa8011b48e81a5961bc3d2c120541418035abc466e4eb73ce7ec7be7a93bc9b679c10bcfb1", "features": [{"input_shapes": [["input_ids", [1, 512]], ["attention_mask", [1, 512]], ["labels", [1, 64]], ["decoder_input_ids", [1, 64]]], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "87fab9d0d49f99dc97d50bbd05a3fd5addb65b2c6bfe6b087a3137792928ff8697484a24dcbac6033fec52b4e98cd1404fa03d5fc111396c9c933102410983c2"}]}, "hf-internal-testing/tiny-random-BertModel": {"model_name_or_path": "hf-internal-testing/tiny-random-BertModel", "model_hash": "e7347098c2715bbf90eb73c3d14e698af0b3ed72650bfdda1510756b7409be8396aeefef5e186f920c0d94c2e1f546e24919e7f39894de4c59e07fdb8e4ce4c9", "features": [{"input_shapes": [["labels", [4]], ["input_ids", [4, 128]], ["token_type_ids", [4, 128]], ["attention_mask", [4, 128]]], "precision": "torch.float32", "num_neuron_cores": 1, "neuron_hash": "0b0150efa64c417ec947fc78f298fc417efea0622966aa493a39d0668933d3073793ba29fc6fb7ba2bd2c4fe234e496076927903ae2fc38017d3ba865523b18d"}]}, "google/vit-base-patch16-224-in21k": {"model_name_or_path": "google/vit-base-patch16-224-in21k", "model_hash": "42ffda2f460d08b51dde8c1c94996b686333142275cab610ae57906cff27f1011aaac36a3d425810fdc84ab9ee93498e6d5a5b8e1163a974a50e7277156ee542", "features": [{"input_shapes": [["pixel_values", [16, 3, 224, 224]], ["labels", [16]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "f0e6fd8a222d4b3a15d9264c797c26a0dc74bcb58838a85d2eb208e183377a2229c51cabf246849487b025f4b6b2f4ef38dad67b9b626550c73e2d9caa682f15"}, {"input_shapes": [["pixel_values", [1, 3, 224, 224]], ["labels", [1]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "5f1133384689e8c214b89c59eb4844f67e0de03215f9dba3abd4d07d0d12f2a7cf90ca36d0912a890b50720450ad289dbc61f3d31afb2069f9c5c8582c1a5dbc"}, {"input_shapes": [["pixel_values", [16, 3, 224, 224]], ["labels", [16]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "c184bf8c51c9fe9d848c0c05571d0a42de417e0a569b3f10f8640694047391563c3b3ca430aecb8e8bd85d20dcf6424b2f3b4d84218725cf3250c44a522541a3"}]}, "PY007/TinyLlama-1.1B-intermediate-step-715k-1.5T": {"model_name_or_path": "PY007/TinyLlama-1.1B-intermediate-step-715k-1.5T", "model_hash": "2f0d5371e180b3eebb6df5eeaf8ed534a63d02ecec5f6453bcce42675221cfb3ab82b99bc90094a9aae440e5d712477b15d507a01ebaf656083a8e5421d34f9b", "features": [{"input_shapes": [["input_ids", [1, 150]], ["attention_mask", [1, 150]], ["labels", [1, 150]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "91853465f9d8fd7c38dc4d696db17b68ebd6e316d0390037f7b4e6f909f49531b1b6cd78f2ffae294b9b4352a4d919e67b7626637d7da390ff3cfb9a66039440"}]}, "NousResearch/Llama-2-7b-hf": {"model_name_or_path": "NousResearch/Llama-2-7b-hf", "model_hash": "e0705a7aca48fd51955f11d7fb5c481bcd9eeba982837347ff2ce1d8bb90599011d6367745b49590dd88cfe8c7c7ad0904ecb35b821408fa33b3d651babd1a7f", "features": [{"input_shapes": [], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "8e87c928371a2ce85ee23b2392c537b64eb2b86ad1612e3d72f44eac2581787d5097e47b0c74f9754543cda1cf0077b7178f93c3e9f0db0774c04ceb497a573c"}]}, "microsoft/beit-base-patch16-224-pt22k-ft22k": {"model_name_or_path": "microsoft/beit-base-patch16-224-pt22k-ft22k", "model_hash": "5f8555ec3b8d727eaff6112f6d9ad27826e0b334481a8ec47c2b36daaf45a12a5ad5d8f0ed5728958ba243c28555e45c9403bf00ba2706cc20ee64f0352f50bb", "features": [{"input_shapes": [["pixel_values", [16, 3, 224, 224]], ["labels", [16]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "d4b9b4409c8d86a44c4b94ae52ad608953ed83b7a6dcdb116afccf985976a2b2761c414e7c3577519402d806efa7c738acfbfef0578fbe79c810569a31a36424"}, {"input_shapes": [["pixel_values", [8, 3, 224, 224]], ["labels", [8]]], "precision": "torch.float32", "num_neuron_cores": 2, "neuron_hash": "b94e517b86964db869d49b7a1ddeceb2889c03b05caddba7f9f6e414e47c3a6c7a5dc7277c11a383b342a4325ffda711a0a72247fd19f9b290e895cbfb5cd7e5"}]}, "philschmid/Llama-2-7b-hf": {"model_name_or_path": "philschmid/Llama-2-7b-hf", "model_hash": "e0705a7aca48fd51955f11d7fb5c481bcd9eeba982837347ff2ce1d8bb90599011d6367745b49590dd88cfe8c7c7ad0904ecb35b821408fa33b3d651babd1a7f", "features": [{"input_shapes": [], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "60e86c81fb56382b64de5b48d49b92d695254cbbe3db011fbb570450dbe472145b65de77121b5a72f8124f55f5a30af7e017c228df1b0527e7dd338515027ec0"}]}, "bert-base-uncased": {"model_name_or_path": "bert-base-uncased", "model_hash": "51ea66ffc9e5a18685bfd4a420cf2469e88494e20f2847484985bcd1c5157a01892305c690f5d23df757c672db4755912f6ec39946ebe941839acd8ef0b3e7e2", "features": [{"input_shapes": [["labels", [16]], ["input_ids", [16, 512]], ["token_type_ids", [16, 512]], ["attention_mask", [16, 512]]], "precision": "torch.bfloat16", "num_neuron_cores": 2, "neuron_hash": "cbcb6f18a21e5e068096768cba9fd02851898b296c3206934dd202b950e60719c7d93c18c2f550c23a6f3ef29d1430c4fe3f2503dce3c84db87b80b05dc85111"}, {"input_shapes": [["labels", [8]], ["input_ids", [8, 512]], ["token_type_ids", [8, 512]], ["attention_mask", [8, 512]]], "precision": "torch.bfloat16", "num_neuron_cores": 2, "neuron_hash": "1e1b75a6a608bd337b2ad44fea44e6cf3a83dfa130bacae5a7c916b73702b1596be05b2289150e55b9c5485c4e5b8b4246b6c4d73f0eb3fd09d71204d5c3314d"}]}, "TinyLlama/TinyLlama-1.1B-Chat-v0.6": {"model_name_or_path": "TinyLlama/TinyLlama-1.1B-Chat-v0.6", "model_hash": "af52d4925feaa72ca214d06a0057935d2e396a299f72a28aa3d1b6a50759bf01fdf2635e9d0adc0f86d2dc6ca5e390a59b0d0e07dadeb26d05700e3377192a0b", "features": [{"input_shapes": [["input_ids", [1, 1024]], ["attention_mask", [1, 1024]], ["labels", [1, 1024]]], "precision": "torch.bfloat16", "num_neuron_cores": 2, "neuron_hash": "29d03bdfa09eedcd5711703ce932180cdd76545b92712d28e8234c5d872ad6121056888ef7978b12faa3a017b599968edbe702f28ce91202e336e189b59da669"}, {"input_shapes": [["input_ids", [1, 1024]], ["attention_mask", [1, 1024]], ["labels", [1, 1024]]], "precision": "torch.bfloat16", "num_neuron_cores": 2, "neuron_hash": "f5a210897c2cc2d487cb8640d35c2dbb3a2b2b2a86d3f604da089445e2be395e30b0553486a623fa97e681047026f9af95861e7f75f1c82456a8a63b42e564ce"}, {"input_shapes": [["input_ids", [2, 1024]], ["attention_mask", [2, 1024]], ["labels", [2, 1024]]], "precision": "torch.bfloat16", "num_neuron_cores": 2, "neuron_hash": "a323cfa789ff7ee78a7930ecabb43afd182dd97e27ec302d2747e36c34fc5d6f4e5a455a69b0b133ce38fdfc97e3456e3b0f748dcbf602981a8ad96304e4cad9"}]}}, "2.11.0.35+4f5279863": {"NousResearch/Llama-2-7b-hf": {"model_name_or_path": "NousResearch/Llama-2-7b-hf", "model_hash": "e0705a7aca48fd51955f11d7fb5c481bcd9eeba982837347ff2ce1d8bb90599011d6367745b49590dd88cfe8c7c7ad0904ecb35b821408fa33b3d651babd1a7f", "features": [{"input_shapes": [], "precision": "torch.float32", "num_neuron_cores": 32, "neuron_hash": "b1376a32545109074892b6229d893fa62766a573a9e32640a5df7ad25d2a5c5d0e961836541fa9a75fe61cc9f49f4e6bc41f0edfd243a6408ffb66c49d64ecd1"}]}}}