onnx-community
/

pyannote-segmentation-3.0

Transformers.js

Model card Files Files and versions Community

Xenova HF staff commited on Jul 12, 2024

Commit

377753f

·

verified ·

1 Parent(s): c548018

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -24,10 +24,10 @@ torch.onnx.export(
     dummy_input,
     'model.onnx',
     do_constant_folding=True,
-    input_names=["input_features"],
     output_names=["logits"],
     dynamic_axes={
-        "input_features": {0: "batch_size", 1: "num_channels", 2: "num_samples"},
         "logits": {0: "batch_size", 1: "num_frames"},
     },
 )

     dummy_input,
     'model.onnx',
     do_constant_folding=True,
+    input_names=["input_values"],
     output_names=["logits"],
     dynamic_axes={
+        "input_values": {0: "batch_size", 1: "num_channels", 2: "num_samples"},
         "logits": {0: "batch_size", 1: "num_frames"},
     },
 )