承德市网站建设_网站建设公司_模板建站_seo优化
2025/12/29 16:27:51 网站建设 项目流程

作为一个小白,今天在看Salesforce/blip-image-captioning-large时发现模型有

model.safetensors
pytorch_model.bin
tf_model.h5

于是见检索了下,做下笔记

权重格式选择:

优先使用 model.safetensors(加载更快、更安全)。
若环境不支持 safetensors,则用 pytorch_model.bin。
若使用 TensorFlow 推理,需保留 tf_model.h5 并替换 model.safetensors。

safetensors 需要安装 pip install safetensors
pytorch_model.bin 需要安装 pip install torch torchvision torchaudio transformers
tf_model.h5 需要安装 pip install tensorflow

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询