OneFlow-ONNX v0.6.0正式发布

OneFlow-ONNX v0.6.0正式发布。新版本提升了转化接口的易用性,开发了多个新特性,并新增支撑6种模型以及20多种算子,此外,还修正了6个转化过程中的bug。 更新详情请查看链接:github.com/Oneflow-Inc…

欢迎通过下列代码一键装置使用,等待你的反应。

pip install oneflow-onnx==0.6.0

以下是本次更新主要内容。

新增支撑的模型

  • 新增支撑MobileNetv3

  • 新增支撑SqueezeNet

  • 新增支撑YOLOv5

  • 新增支撑Disco Diffusion

  • 新增支撑LiBai仓库的T5模型

  • 新增支撑LiBai仓库的VisionTransformer 模型

新增支撑的Op

  • 新增支撑hard_swish和hard_sigmoid op

  • 新增支撑arange op

  • 新增支撑expand_dims op

  • 新增支撑narrow op

  • 新增支撑silu op

  • 新增支撑upsample_nearest_2d op

  • 新增支撑var op

  • 新增支撑conv1d op

  • 新增支撑scalar_div op

  • 新增支撑cublas_fused_mlp op

  • 新增支撑elementwise max/min op

  • 新增支撑broadcast_matmul,where,scalar_logical_less,scalar_logical_greater,gather op

  • 新增expand op

  • 新增支撑fill_ op

  • 新增支撑gelu op

  • 新增支撑layernorm op

  • 新增支撑amp_white_identity/amp_black_identity op

BUG 修正

  • 修正装置oneflow_onnx时不自动装置onnx、onnxruntime等依靠包的问题

  • 修正因为版本更新导致的maxpool op转化失利的bug

  • 修正unsqueeze op在Opset 13下的bug

  • 修正获取tensor时硬编码导致过错

  • 修正保存超大tensor时的size推导过错

  • 修正了在LiBai下, 用T5做test转ONNX时遇到的问题。(支撑了OneFlow在编译Graph时,采用Global Tensor进行推理的写法以及flow.bool类型Tensor作为输入)

  • 修正 pool 多了一个index 输出导致TensorRT推理失利的问题

新增Feature

  • 重构代码仓库的示例文档

  • 重构导出onnx api,flow_weight_dir参数可选,提升易用性

  • 允许mapping过程中访问到原始op_node以取得更多必要信息

  • CI支撑black格式化

  • 支撑Graph里边有Free Eager Tensor

  • 转化ONNX时支撑多个输入的Graph对象

版本发布过程中,感谢以下贡献者的支撑 :@Flowingsun007 、@doombeaker 、@liujuncheng @leaves-zwx @CPFLAME、@BBuf、@zhongshsh

欢迎下载体验 OneFlow v0.8.0 最新版本:
github.com/Oneflow-Inc…