optimum-neuron-cache / registry.json
michaelbenayoun's picture
Add hf-internal-testing/tiny-random-gpt_neo in registry for NeuronHash 6b09ba88e4bc3f63d55273e8ceefaddb2c54f9d63c2b1b4ee0e24e0c92cbb92fb1e45f41832723c680f42ee202b7f232235531d13ae8ac547531c38b07517c97
bcb5a95
raw
history blame
3.01 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"}]}}}