Skip to content

Releases: Oneflow-Inc/oneflow_convert

v0.6.0

25 Oct 13:25
55f3019
Compare
Choose a tag to compare

v0.6.0发布

v0.6.0版本相比于v0.5.4版本提升了转换接口的易用性,开发了多个新的特性。并新增了6种模型的支持以及20余种Op的支持,以及修复6个转换过程中的bug。版本发布过程中,感谢以下贡献者的支持。 @Flowingsun007 @doombeaker @liujuncheng @leaves-zwx @CPFLAME @BBuf @zhongshsh

pip install oneflow-onnx==0.6.0 一键安装使用 🎉🎉🎉 。

新增支持的模型

  • 新增支持 mobilenetv3。#63
  • 新增支持 squeezenet 。#63
  • 新增支持 yolov5。#95
  • 新增支持 disco diffusion。#94
  • 新增支持 libai 仓库的 T5 模型。#92
  • 新增支持 libai 仓库的 VisionTransformer 模型。#103

新增支持的Op

  • 新增支持hard_swish和hard_sigmoid op。#63
  • 新增支持arange op。#71
  • 新增支持 expand_dims op。#74
  • 新增支持 narrow op。#75
  • 新增支持 silu op。#78
  • 新增支持 upsample_nearest_2d op。#79
  • 新增支持 var op。#81
  • 新增支持 conv1d op。#83
  • 新增支持 scalar_div op。#84
  • 新增支持 cublas_fused_mlp op。#90
  • 新增支持 elementwise max/min op。#92
  • 新增支持 broadcast_matmul,where,scalar_logical_less,scalar_logical_greater,gather op。#93
  • 新增 expand op。#95
  • 新增支持 fill_ op。#96
  • 新增支持gelu op。#100
  • 新增支持layernorm op。#101
  • 新增支持amp_white_identity/amp_black_identity op。#102

BUG 修复

  • 修复安装oneflow_onnx时不自动安装onnx,onnxruntime等依赖包的问题。#60
  • 修复因为版本更新导致的maxpool op转换失败的bug。#61
  • 修复unsqueeze op在Opset 13下的bug。#65
  • 修复获取tensor时硬编码导致错误。#77
  • 修复保存超大tensor时的size推导错误。#82
  • 修复了在libai下, 用T5做test 转onnx时遇到的问题。(支持了oneflow在编译graph时,采用global tensor进行推理的写法以及flow.bool类型Tensor作为输入)#88
  • 修复 pool 多了一个index 输出导致TensorRT推理失败的问题。#97

新Feature

  • 重构代码仓库的示例文档。#68
  • 重构导出onnx api,flow_weight_dir参数可选,提升易用性。#73
  • 允许mapping过程中访问到原始op_node以获得更多必要信息。#80
  • CI支持black 格式化。#86
  • 支持Graph里面有Free Eager Tensor。#89
  • 转换onnx时支持多个输入的Graph对象。#99