Granite Timeseries Patchtsmixer

🚀 Transformers.js

Transformers.js 是一個與 ONNX 權重兼容的庫，它基於 ibm-granite/granite-timeseries-patchtsmixer 基礎模型，可用於時間序列預測。

🚀 快速開始

本庫將 ibm-granite/granite-timeseries-patchtsmixer 模型的 ONNX 權重進行適配，以兼容 Transformers.js。

📦 安裝指南

如果你還沒有安裝 Transformers.js JavaScript 庫，可以使用以下命令從 NPM 進行安裝：

npm i @huggingface/transformers

💻 使用示例

基礎用法

以下是使用 onnx-community/granite-timeseries-patchtsmixer 進行時間序列預測的示例：

import { PatchTSMixerForPrediction, Tensor } from '@huggingface/transformers';

const model_id = "onnx-community/granite-timeseries-patchtsmixer";
const model = await PatchTSMixerForPrediction.from_pretrained(model_id, { dtype: "fp32" });

const dims = [64, 512, 7];
const prod = dims.reduce((a, b) => a * b, 1);
const past_values = new Tensor('float32',
    Float32Array.from({ length: prod }, (_, i) => i / prod),
    dims,
);
const { prediction_outputs } = await model({ past_values });
console.log(prediction_outputs);

📚 詳細文檔

注意事項

目前為 ONNX 權重單獨創建一個倉庫是一個臨時解決方案，直到 WebML 獲得更多關注。如果你想讓你的模型適用於 Web，我們建議使用 🤗 Optimum 將模型轉換為 ONNX 格式，並像本倉庫一樣組織你的倉庫結構（將 ONNX 權重放在名為 onnx 的子文件夾中）。

📄 許可證

文檔中未提及許可證信息。