Upload ONNX model for Nvidia L40S
#1
by robustdev - opened
Original ONNX Model
==== ONNX Model Summary ====
Model: onnx_models/model.onnx
IR Version: 9
Producer: pytorch 2.9.1
Opset: ai.onnx:20
== Inputs ==
input_ids: shape=(batch, seq), dtype=INT64
attention_mask: shape=(batch, seq), dtype=INT64
== Outputs ==
logits: shape=(batch, 2), dtype=FLOAT
Num Nodes: 2575
Num Initializers: 138
Gerald001 changed pull request status to merged