fengzii/granite-timeseries-ttm-r2-npu
模型介绍
文件和版本
Pull Requests
讨论
分析

granite-timeseries-ttm-r2 昇腾 NPU 适配

环境信息

  • 硬件: Ascend 910 (Ascend910_9362)
  • 软件: CANN 8.5.1, PyTorch 2.9.0 + torch_npu 2.9.0.post1, Python 3.11
  • 模型: ibm-granite/granite-timeseries-ttm-r2 (表格数据)
  • 精度: float16

适配步骤

export HF_ENDPOINT=https://hf-mirror.com
pip install transformers granite-tsfm torch_npu numpy
python3 inference.py

验证结果

  • 推理: ✅ 测试通过
  • 验证日期: 2026-08-20

原始模型

  • HuggingFace: https://huggingface.co/ibm-granite/granite-timeseries-ttm-r2

NPU 验证结果

  • 架构: TinyTimeMixerForPrediction | 加载器: TinyTimeMixer | 耗时: 120s
  • 硬件: Ascend 910 (Ascend910_9362), CANN 8.5.1, torch_npu 2.9.0
  • 验证: ✅ 随机输入 forward 通过 | 日期: 2026-08-23
  • 输出样本: 仓库: granite-timeseries-ttm-r2-npu | 大小: 0.0GB | 耗时: 22s | 结果: OK
  • 验证截图: assets/