如何使用别人开发好的任何模型
简要描述
Hugging Face 平台支持多种开源模型架构和文件格式,以下是常见模型类型及其对应的文件后缀整理(已修正部分笔误并补充完整信息):
1. Transformer 系列模型
主要用途:NLP 任务(BERT、GPT、T5 等)
常见后缀:
.bin
(PyTorch 权重文件).safetensors
(Hugging Face 安全格式,替代.bin
).h5
(TensorFlow/Keras 权重文件).msgpack
(旧版 Flax/JAX 格式).gguf
(量化格式,用于 llama.cpp 等推理工具)
2. Diffusion 模型(文生图、图生图)
代表模型:Stable Diffusion、DALL-E
常见后缀:
.safetensors
(推荐的安全格式,替代.ckpt
).ckpt
(PyTorch 检查点文件,逐渐被弃用).onnx
(导出为跨平台推理格式)
3. 卷积神经网络 (CNN)
主要用途:图像分类、目标检测
常见后缀:
.pth
(PyTorch 原生格式).h5
(TensorFlow/Keras 格式).onnx
(标准化推理格式).tflite
(移动端轻量化模型)
4. 语音模型(ASR、TTS)
代表模型:Whisper、Wav2Vec2
常见后缀:
.bin
(PyTorch).safetensors
.onnx
(导出为跨框架格式)
5. 多模态模型
代表模型:CLIP、FLAVA
常见后缀:
.bin
、.safetensors
(PyTorch).h5
(TensorFlow)
6. 其他特殊格式
量化/轻量化格式:
.gguf
(llama.cpp、alpaca.cpp 使用).ptl
(PyTorch Mobile)
框架通用格式:
.onnx
(支持跨框架推理).pb
(TensorFlow Protocol Buffer)
Last updated