掘金 人工智能 前天 00:48
【Hugging Face】Hugging Face模型的基本使用
index_new5.html
../../../zaker_core/zaker_tpl_static/wap/tpl_guoji1.html

 

本文详细介绍了Hugging Face模型的使用方法,主要分为两种途径:通过huggingface_hub进行统一封装调用,以及利用transformers和diffusers库进行特定任务处理。文章以Qwen3-0.6B文本生成和FLUX.1-dev绘图为例,展示了如何通过huggingface_hub接入API并实现文本生成和图像生成。同时,文章也简要介绍了如何使用transformers库进行情感分析,以及使用diffusers库进行sdxl-turbo模型的图像生成。整体内容旨在帮助用户快速上手Hugging Face模型,实现AI应用开发。

🚀 Hugging Face模型可通过huggingface_hub和transformers/diffusers两大途径使用。huggingface_hub提供统一封装,需API Key;transformers专注于NLP和多模态任务,diffusers则侧重于图像、音频等扩散模型生成,两者均无需API Key。

💡 通过huggingface_hub使用模型,以Qwen3-0.6B文本生成为例,展示了在Colab环境中,通过`InferenceClient`接入模型,并提供代码示例,说明如何进行英文和中文的文本生成,并展示了FLUX.1-dev模型的文本到图像生成过程。

💻 文章介绍了如何使用transformers库中的`pipeline`功能进行情感分析,以`distilbert-base-uncased-finetuned-sst-2-english`模型为例,展示了对正负面评价的分析过程,并提供了具体的代码实现。

🖼️ 对于图像生成,文章详细演示了如何使用diffusers库中的`DiffusionPipeline`加载`stabilityai/sdxl-turbo`模型,并通过设置prompt、negative_prompt、num_inference_steps等参数,生成指定图像,并展示了代码及生成效果。

前言

前面我们了解了Hugging Face Space空间 和 DataSets数据集 的基本使用方式,今天我们继续探索Hugging Face Models的基本使用方式。对往期内容感兴趣的小伙伴也可以看往期:

模型使用方式

使用Hugging Face模型有两种方式,一种是通过 Hugging Face提供的 huggingface_hub,一种是通过Hugging Face提供的 transformers 和 diffusers

通过huggingface_hub使用

由于个人电脑配置有限,我这里在Google的Colab上使用,在本地使用步骤也是类似的,对Google Colab还不了解的小伙伴可以看往期:
初识Google Colab

Qwen3-0.6B文本生成

以Qwen3-0.6B模型为例,进入模型详情

点击【View Code Snippets】查看接入代码

在Colab上新建Colab Notebooks,输入代码

import osfrom huggingface_hub import InferenceClient# Set the environment variable for the current processos.environ['HF_TOKEN'] = "your hugging face api key"client = InferenceClient(    provider="featherless-ai",    api_key=os.environ["HF_TOKEN"],)completion = client.chat.completions.create(    model="Qwen/Qwen3-0.6B",    messages=[        {            "role""user",            "content""What is the capital of France?"        }    ],)print(completion.choices[0].message.content)

点击左侧运行,输出结果如下

这里是英文的,我们也可以AI输出中文结果,稍微改动一下提示词

import osfrom huggingface_hub import InferenceClient# Set the environment variable for the current processos.environ['HF_TOKEN'] = "your hugging face api key"client = InferenceClient(    provider="featherless-ai",    api_key=os.environ["HF_TOKEN"],)completion = client.chat.completions.create(    model="Qwen/Qwen3-0.6B",    messages=[        {            "role""user",            "content""默认使用中文回复"        },        {            "role""user",            "content""What is the capital of France?"        }    ],)print(completion.choices[0].message.content)

此时AI就会以中文回复了

FLUX.1-dev绘图

进入FLUX.1-dev模型详情

点击【View Code Snippets】查看接入代码

在Colab文档上新建一个代码块,输入以下代码

import osfrom huggingface_hub import InferenceClientfrom IPython.display import displayos.environ['HF_TOKEN'] = "your api key"client = InferenceClient(    provider="nebius",    api_key=os.environ["HF_TOKEN"],)# output is a PIL.Image objectimage = client.text_to_image(    "Astronaut riding a horse",    model="black-forest-labs/FLUX.1-dev",)# 展示图片display(image)

点击运行,执行结束后,我们将得到一张图片

通过transformers和diffusers使用

transformers 和 diffusers 是Hugging Face提供的两个重要Python库,使用较为复杂,这里只做模型的简单使用,不深入了解。

情感分析

以distilbert-base-uncased-finetuned-sst-2-english模型为例,进入distilbert-base-uncased-finetuned-sst-2-english模型详情,复制模型名称

在Colab中创建代码块,添加如下代码

from transformers import pipeline# huggingface远程模型名model_name = "distilbert-base-uncased-finetuned-sst-2-english"# 创建管道classifier = pipeline("sentiment-analysis", model=model_name)# 进行情感分析result = classifier("I hate you")[0]print(result)result = classifier("I love you")[0]print(result)

点击左侧运行,输出结果如下

sdxl-turbo绘图

Diffusers库,这里以stabilityai/sdxl-turbo模型为例,进入stabilityai/sdxl-turbo模型详情,复制模型名称

在Colab中创建代码块,添加如下代码

from diffusers import DiffusionPipelineimport torchif torch.cuda.is_available():    device"cuda"    torch_dtype = torch.float16else:    device"cpu"    torch_dtype = torch.float32pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch_dtype)pipe.to(device)image = pipe(    prompt="Astronaut riding a horse",    negative_prompt="low quality, bad quality, loss body"# 反向提示词    num_inference_steps=1# 步长    guidance_scale=0.0# 引导系数    width=512,    height=512).images[0]display(image)

等待执行,执行完成后,可以看到生成图片的效果如下:

参考

友情提示

见原文:【Hugging Face】Hugging Face模型的基本使用

本文同步自微信公众号 "程序员小溪" ,这里只是同步,想看及时消息请移步我的公众号,不定时更新我的学习经验。

Fish AI Reader

Fish AI Reader

AI辅助创作,多种专业模板,深度分析,高质量内容生成。从观点提取到深度思考,FishAI为您提供全方位的创作支持。新版本引入自定义参数,让您的创作更加个性化和精准。

FishAI

FishAI

鱼阅,AI 时代的下一个智能信息助手,助你摆脱信息焦虑

联系邮箱 441953276@qq.com

相关标签

Hugging Face AI模型 transformers diffusers 机器学习
相关文章