w
gcw_uQ09W7jl/borzoi-human-NPU
模型介绍
文件和版本
Pull Requests
讨论
分析

multimolecule/borzoi-human on Ascend NPU

模型与任务

  • 模型：multimolecule/borzoi-human
  • Revision：6a377a3bf83947284f601ddd57a7ed27b370e4f0
  • 适配路线：long-context-tracks
  • 设备：npu:0

环境依赖

{
  "torch": "2.9.0+cpu",
  "torch_npu": "2.9.0.post1+gitee7ba04",
  "transformers": "4.57.6",
  "python": "3.11.14",
  "device_name": "Ascend910_9362",
  "npu_count": 2,
  "model_dir": "persist_models/multimolecule_borzoi-human"
}

安装：pip install -r requirements.txt

NPU 推理

默认执行：python inference.py

真实输入

{
  "sequence": "ACTCCCCTGCCCTCAACAAGATGTTTTGCCAACTGGCCAAGACCTGCCCTGTGCAGCTGTGGGTTGATTCCACACCCCCGCCCGGCACCCGCGTCCGCGCCATGGCCATCTACAAGCAGT...",
  "sequence_length": 524288,
  "sha256": "485bdeec7800c9614a28f304de8dd98c5a1d82257e68dc22d74f1ae2414abb05",
  "alphabet": "A/C/G/T/N",
  "direction": "5'->3'",
  "species": "human",
  "target_length": 6144,
  "num_tracks": 7611,
  "bin_size_bp": 32,
  "tokenizer": "DnaTokenizer",
  "vocab": [
    "A",
    "C",
    "G",
    "T",
    "N"
  ]
}

真实推理结果

{
  "logits_shape": [
    1,
    6144,
    7611
  ],
  "coverage_shape": [
    1,
    6144,
    7611
  ],
  "logits_dtype": "torch.float32",
  "coverage_dtype": "torch.float32",
  "logits_min": -9.580877304077148,
  "logits_max": 21.37462615966797,
  "logits_mean": -1.481689214706421,
  "logits_sample_0_0": [
    -1.720968246459961,
    -1.7014912366867065,
    -3.3326332569122314,
    -3.5460689067840576,
    -3.2910983562469482
  ],
  "coverage_mean": 0.738118052482605,
  "finite_ratio": 100.0
}

CPU-NPU 一致性

{
  "max_abs_diff": 0.00940704345703125,
  "mean_abs_diff": 0.00023349769569531604,
  "atol": 0.02,
  "rtol": 0.001,
  "pearson_corr_subset": 0.9999999918280094,
  "passed": true,
  "finite_ok": true
}

性能

{
  "warmup_runs": 3,
  "stable_runs": 10,
  "avg_ms": 236.57,
  "min_ms": 231.98,
  "max_ms": 274.74,
  "p50_ms": 232.41,
  "windows_per_s": 4.23,
  "bases_per_s": 2216191,
  "input_length": 524288,
  "output_shape": [
    1,
    6144,
    7611
  ],
  "dtype": "float32",
  "device": "npu:0"
}

自验证截图

Agent workflow

NPU device call

Model result

三张图片均由 xterm.js 根据本次真实日志生成。

已知限制

  • 仅对本次记录的模型 revision、输入契约、dtype 与 Ascend 环境完成验证。

标签

#NPU #Ascend #Ascend910