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…