Add prajjwal1/bert-tiny in registry for NeuronHash 578b3c544f3df77ff6f9a15342a2edb85638d68a7528278ddc6c51dc4bcae528409d7a3f799f6934a87934d0489cd8d6c893b2ab33c9e7da16ddd5442af917c3

registry.json

@@ -1 +1 @@
-{"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.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"}]}}}