第二章 Hugging Face简介

1,174 阅读1分钟

第2章 Hugging Face简介

2.1 Hugging Face核心功能介绍

  • 模型、数据集存储库:Hugging Face Hub,主要用于托管模型和数据集,并详细描述模型名称、分类标签、模型的开源协议、arXiv.org的论文引用等。
  • 模型卡片:包括模型用途、制作模型的背景、模型的详细介绍、引用论文、使用说明、推理API等。
  • 推理API:文字形式的输入输出、上传图片和调用游览器获取麦克风实时音频。
  • Space应用:提供创建和部署机器学习应用的功能。

2.2 Hugging Face开源库

  • Transformers:下载和训练SOTA的预训练模型,支持PyTorch、TensorFlow和JAX。
  • Datasets:使用代码方法,轻松加载各种数据集。
  • Diffusers:扩散模型工具箱,使用各种噪声调度器,调节模型推理过程中的生成速度和质量。
  • Accelerate:支持在任何类型的设备上运行原本PyTorch训练脚本。
  • Optimum:Transformers扩展,性能优化工具。
  • timm:深度学习库,包含图像模型、优化器、调度器以及训练/验证脚本等。

2.3 Gradio工具介绍

Gradio用于构建机器学习和数据科学演示以及Web应用,能够通过浏览器拖放图片、粘贴文本、录制声音并进行演示互动。

import gradio as gr

def greet(name):
    return "Hello " + name + "!"

demo = gr.Interface(fn=greet, inputs="text", outputs="text")

demo.launch()

image.png