Upload ONNX model files
#6
by sarah-cisco - opened
ONNX Model
==== ONNX Model Summary ====
Model: /workspace/onnx_models/toxicity_onnx.onnx
IR Version: 10
Producer: pytorch 2.11.0+cu130
Opset: ai.onnx:20
== Inputs ==
input_ids: shape=(batch, seq), dtype=INT64
attention_mask: shape=(batch, seq), dtype=INT64
== Outputs ==
logits: shape=(batch, 11), dtype=FLOAT
Num Nodes: 2650
Num Initializers: 371
sarah-cisco changed pull request status to closed