packages/core/src/architectures/moonshine.js (264 lines of code) (raw):

// NOTE: This file has been auto-generated. Do not edit directly. export default { model_type: "moonshine", models: [ { model_id: "hf-internal-testing/tiny-random-MoonshineForConditionalGeneration", dtype: "fp32", architectures: ["MoonshineForConditionalGeneration"], ops: [ "Add", "Cast", "Concat", "Constant", "ConstantOfShape", "Conv", "Cos", "Div", "Equal", "Erf", "Expand", "Gather", "GatherND", "Greater", "If", "InstanceNormalization", "Less", "MatMul", "Mul", "Neg", "NonZero", "Pow", "Range", "ReduceMax", "ReduceMean", "Reshape", "Shape", "Sigmoid", "Sin", "Slice", "Softmax", "Sqrt", "Sub", "Tanh", "Transpose", "Trilu", "Unsqueeze", "Where", ], }, { model_id: "onnx-community/moonshine-base-ONNX", dtype: "fp16", architectures: ["MoonshineForConditionalGeneration"], ops: [ "Add", "Cast", "Concat", "Constant", "ConstantOfShape", "Conv", "Cos", "Div", "Equal", "Erf", "Expand", "Gather", "Greater", "If", "InstanceNormalization", "MatMul", "Mul", "Neg", "Pow", "Range", "ReduceMean", "Reshape", "Shape", "Sigmoid", "Sin", "Slice", "Softmax", "Sqrt", "Sub", "Tanh", "Transpose", "Trilu", "Unsqueeze", "Where", ], }, { model_id: "onnx-community/moonshine-base-ONNX", dtype: "quantized", architectures: ["MoonshineForConditionalGeneration"], ops: [ "Add", "Cast", "Concat", "Constant", "ConstantOfShape", "Conv", "Cos", "DequantizeLinear", "Div", "DynamicQuantizeLinear", "Equal", "Erf", "Expand", "Gather", "Greater", "If", "InstanceNormalization", "MatMul", "MatMulInteger", "Mul", "Neg", "Pow", "Range", "ReduceMean", "Reshape", "Shape", "Sigmoid", "Sin", "Slice", "Softmax", "Sqrt", "Sub", "Tanh", "Transpose", "Trilu", "Unsqueeze", "Where", ], }, { model_id: "onnx-community/moonshine-base-ONNX", dtype: "int8", architectures: ["MoonshineForConditionalGeneration"], ops: [ "Add", "Cast", "Concat", "Constant", "ConstantOfShape", "ConvInteger", "Cos", "DequantizeLinear", "Div", "DynamicQuantizeLinear", "Equal", "Erf", "Expand", "Gather", "Greater", "If", "InstanceNormalization", "MatMul", "MatMulInteger", "Mul", "Neg", "Pow", "Range", "ReduceMean", "Reshape", "Shape", "Sigmoid", "Sin", "Slice", "Softmax", "Sqrt", "Sub", "Tanh", "Transpose", "Trilu", "Unsqueeze", "Where", ], }, { model_id: "onnx-community/moonshine-base-ONNX", dtype: "q4f16", architectures: ["MoonshineForConditionalGeneration"], ops: [ "Add", "Cast", "Concat", "Constant", "ConstantOfShape", "Conv", "Cos", "Div", "Equal", "Erf", "Expand", "Gather", "Greater", "If", "InstanceNormalization", "MatMul", "MatMulNBits", "Mul", "Neg", "Pow", "Range", "ReduceMean", "Reshape", "Shape", "Sigmoid", "Sin", "Slice", "Softmax", "Sqrt", "Sub", "Tanh", "Transpose", "Trilu", "Unsqueeze", "Where", ], }, { model_id: "onnx-community/moonshine-base-ONNX", dtype: "bnb4", architectures: ["MoonshineForConditionalGeneration"], ops: [ "Add", "Cast", "Concat", "Constant", "ConstantOfShape", "Conv", "Cos", "Div", "Equal", "Erf", "Expand", "Gather", "Greater", "If", "InstanceNormalization", "MatMul", "MatMulBnb4", "Mul", "Neg", "Pow", "Range", "ReduceMean", "Reshape", "Shape", "Sigmoid", "Sin", "Slice", "Softmax", "Sqrt", "Sub", "Tanh", "Transpose", "Trilu", "Unsqueeze", "Where", ], }, ], };