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",
],
},
],
};