OneFlow v0.9.0正式发布

OneFlow v0.9.0正式发布,加强PyTorch API和模型的兼容性,新增86个API接口和算子。支持PyTorch模型一键迁移,包括Stable Diffusion、GLM、YOLOv5等。此外,引入自动并行功能,性能和内存优化显著,提供更好的调试工具和友好的错误提示。OneFlow-ONNX工具也得到升级,支持更多模型和Op。
摘要由CSDN通过智能技术生成

24cba4dda2d927dcaa7091d83ffd734a.png


今天是 OneFlow 开源的第 903 天,OneFlow v0.9.0 正式发布。本次更新包含 640 个 commit,完整更新列表请查看链接:https://github.com/Oneflow-Inc/oneflow/releases/tag/v0.9.0,欢迎下载体验新版本,期待你的反馈。

OneFlow v0.9.0 主要包括以下新增亮点功能和优化:

1. PyTorch 兼容性

OneFlow 对 PyTorch API 和 模型的兼容性进一步提升。新增与 PyTorch 对齐的 86 个 API 接口 和 算子, 修复了 104 个算子兼容性相关的 bug。支持更多的 PyTorch 模型一键迁移成 OneFlow 模型,并获得性能加速。

  • 支持 Stable DiffusionGLMYOLOv5 等模型一键迁移为 OneFlow 模型

  • oneflow.load 支持直接加载 torch.save 的模型,模型迁移更方便

评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值