OFA-Sys / Chinese-CLIP

Chinese version of CLIP which achieves Chinese cross-modal retrieval and representation generation.
MIT License
4.5k stars 464 forks source link

怎样将该模型转换成pytorchscript格式的模型? #127

Closed zhangjh closed 1 year ago

zhangjh commented 1 year ago

我是一个后端工程师,对AI不算很熟,请问如何将你们的模型转换成torchscript格式的模型?我需要部署在手机端,目前只找到支持这个格式的方案。

yangapku commented 1 year ago

您好,目前在部署侧我们还没有支持torchscript格式,支持的是ONNX和tensorRT格式的部署方案。未来如果有精力我们会优先考虑这个需求哈。

zhangjh commented 1 year ago

好的感谢,我研究下onnx的