Oneflow-Inc / libai

LiBai(李白): A Toolbox for Large-Scale Distributed Parallel Training
https://libai.readthedocs.io
Apache License 2.0
389 stars 55 forks source link

fix loader #504

Closed QiaolingChen00 closed 1 year ago

QiaolingChen00 commented 1 year ago

最新 OneFlow 生成的模型权重是文件而非文件夹,所以需要在libai load 模型时,对 模型path 是否是文件夹路径 的判断语句进行修改。