Upload ONNX model for Nvidia RTX PRO 6000
#2
by sarah-cisco - opened
Original ONNX Model
==== ONNX Model Summary ====
Model: model.onnx
IR Version: 10
Producer: pytorch 2.11.0+cu130
Opset: ai.onnx:20
== Inputs ==
input_ids: shape=(batch_size, sequence_length), dtype=INT64
attention_mask: shape=(batch_size, sequence_length), dtype=INT64
== Outputs ==
logits: shape=(batch_size, 12), dtype=FLOAT
Num Nodes: 2656
Num Initializers: 371
sarah-cisco changed pull request status to merged