packages/core/src/architectures/longt5.js (348 lines of code) (raw):
// NOTE: This file has been auto-generated. Do not edit directly.
export default {
model_type: "longt5",
models: [
{
model_id: "Xenova/long-t5-encodec-tglobal-base",
dtype: "quantized",
architectures: ["LongT5Model"],
ops: [
"Abs",
"Add",
"And",
"Cast",
"Concat",
"Constant",
"ConstantOfShape",
"CumSum",
"DequantizeLinear",
"Div",
"DynamicQuantizeLinear",
"Einsum",
"Equal",
"Expand",
"Floor",
"Gather",
"Greater",
"GreaterOrEqual",
"Identity",
"Less",
"LessOrEqual",
"Log",
"MatMul",
"MatMulInteger",
"Min",
"Mod",
"Mul",
"Neg",
"Not",
"OneHot",
"Pad",
"Pow",
"Range",
"ReduceMax",
"ReduceMean",
"ReduceSum",
"Reshape",
"Shape",
"Slice",
"Softmax",
"Sqrt",
"Sub",
"Tanh",
"Tile",
"Transpose",
"Unsqueeze",
"Where",
],
},
{
model_id: "Xenova/long-t5-encodec-tglobal-base",
dtype: "fp32",
architectures: ["LongT5Model"],
ops: [
"Abs",
"Add",
"And",
"Cast",
"Concat",
"Constant",
"ConstantOfShape",
"CumSum",
"Div",
"Einsum",
"Equal",
"Expand",
"Floor",
"Gather",
"Greater",
"GreaterOrEqual",
"Identity",
"Less",
"LessOrEqual",
"Log",
"MatMul",
"Min",
"Mod",
"Mul",
"Neg",
"Not",
"OneHot",
"Pad",
"Pow",
"Range",
"ReduceMax",
"ReduceMean",
"ReduceSum",
"Reshape",
"Shape",
"Slice",
"Softmax",
"Sqrt",
"Sub",
"Tanh",
"Tile",
"Transpose",
"Unsqueeze",
"Where",
],
},
{
model_id: "Xenova/long-t5-local-base",
dtype: "fp16",
architectures: ["LongT5ForConditionalGeneration"],
ops: [
"Add",
"And",
"Cast",
"Concat",
"Constant",
"ConstantOfShape",
"Div",
"Einsum",
"Expand",
"Gather",
"Greater",
"If",
"Less",
"LessOrEqual",
"Log",
"MatMul",
"Min",
"Mod",
"Mul",
"Neg",
"Pad",
"Pow",
"Range",
"ReduceMean",
"Reshape",
"Shape",
"Slice",
"Softmax",
"Sqrt",
"Sub",
"Tanh",
"Tile",
"Transpose",
"Unsqueeze",
"Where",
],
},
{
model_id: "Xenova/long-t5-local-base",
dtype: "quantized",
architectures: ["LongT5ForConditionalGeneration"],
ops: [
"Add",
"And",
"Cast",
"Concat",
"Constant",
"ConstantOfShape",
"DequantizeLinear",
"Div",
"DynamicQuantizeLinear",
"Einsum",
"Expand",
"Gather",
"Greater",
"If",
"Less",
"LessOrEqual",
"Log",
"MatMul",
"MatMulInteger",
"Min",
"Mod",
"Mul",
"Neg",
"Pad",
"Pow",
"Range",
"ReduceMean",
"Reshape",
"Shape",
"Slice",
"Softmax",
"Sqrt",
"Sub",
"Tanh",
"Tile",
"Transpose",
"Unsqueeze",
"Where",
],
},
{
model_id: "Xenova/long-t5-tglobal-base-16384-book-summary",
dtype: "fp16",
architectures: ["LongT5ForConditionalGeneration"],
ops: [
"Abs",
"Add",
"And",
"Cast",
"Concat",
"Constant",
"ConstantOfShape",
"CumSum",
"Div",
"Einsum",
"Equal",
"Expand",
"Floor",
"Gather",
"Greater",
"GreaterOrEqual",
"Identity",
"If",
"Less",
"LessOrEqual",
"Log",
"MatMul",
"Min",
"Mod",
"Mul",
"Neg",
"Not",
"OneHot",
"Pad",
"Pow",
"Range",
"ReduceMax",
"ReduceMean",
"ReduceSum",
"Reshape",
"Shape",
"Slice",
"Softmax",
"Sqrt",
"Sub",
"Tanh",
"Tile",
"Transpose",
"Unsqueeze",
"Where",
],
},
{
model_id: "Xenova/long-t5-tglobal-base-16384-book-summary",
dtype: "quantized",
architectures: ["LongT5ForConditionalGeneration"],
ops: [
"Abs",
"Add",
"And",
"Cast",
"Concat",
"Constant",
"ConstantOfShape",
"CumSum",
"DequantizeLinear",
"Div",
"DynamicQuantizeLinear",
"Einsum",
"Equal",
"Expand",
"Floor",
"Gather",
"Greater",
"GreaterOrEqual",
"Identity",
"If",
"Less",
"LessOrEqual",
"Log",
"MatMul",
"MatMulInteger",
"Min",
"Mod",
"Mul",
"Neg",
"Not",
"OneHot",
"Pad",
"Pow",
"Range",
"ReduceMax",
"ReduceMean",
"ReduceSum",
"Reshape",
"Shape",
"Slice",
"Softmax",
"Sqrt",
"Sub",
"Tanh",
"Tile",
"Transpose",
"Unsqueeze",
"Where",
],
},
{
model_id: "onnx-internal-testing/tiny-random-LongT5Model-ONNX",
dtype: "fp32",
architectures: ["LongT5Model"],
ops: [
"Add",
"And",
"Cast",
"Concat",
"Constant",
"ConstantOfShape",
"Div",
"Einsum",
"Equal",
"Expand",
"Gather",
"Greater",
"Identity",
"If",
"Less",
"Log",
"MatMul",
"Min",
"Mod",
"Mul",
"Neg",
"Pad",
"Pow",
"Range",
"ReduceMean",
"Relu",
"Reshape",
"ScatterND",
"Shape",
"Slice",
"Softmax",
"Sqrt",
"Sub",
"Transpose",
"Trilu",
"Unsqueeze",
"Where",
],
},
],
};