optimum-neuron-cache / registry.json
michaelbenayoun's picture
Add Helsinki-NLP/opus-mt-en-ro in registry for NeuronHash 4e3254e4f824724d529aa8d7aafd0624f762f7efabaaf06b249e4d9181109285f5beabfaa02c42d5c924659e069467af7ea35227ea31995b5370d5ab6ac7ef2b
6524ef7
raw
history blame
1.16 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"}]}}}