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