Update export_meta.py

err id for export embed onnx
This commit is contained in:
JIJIN CHEN 2024-06-21 16:05:10 +08:00 committed by GitHub
parent 362eed972c
commit 7649c9ef59
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -53,7 +53,7 @@ class ContextualEmbedderExport(torch.nn.Module):
0: "num_hotwords",
},
"hw_embed": {
0: "num_hotwords",
1: "num_hotwords",
},
}