一步一步带你实战 FLUX.1 文生图大模型,在线体验一键出图

B站影视 2025-01-16 06:58 1

摘要:上期图文教程,我们分享了flux.1 文生图大模型以及flux.1 的最新工具集,但是有网友反馈代码运行提示错误,其实代码并没有问题,只是基于 hugging face 的第三方库需要获取相关的权限,然后代码才能正常下载对应的模型权重,本期我们首先先从头介绍一

flux文生图模型

上期图文教程,我们分享了 flux.1 文生图大模型以及flux.1 的最新工具集,但是有网友反馈代码运行提示错误,其实代码并没有问题,只是基于 hugging face 的第三方库需要获取相关的权限,然后代码才能正常下载对应的模型权重,本期我们首先先从头介绍一下如何使用代码,并能够正常出图。

flux.1 文生图模型

首先第一步,我们需要登录 hugging face 的官方网站,并找到 black-forest-labs 官方地址,点击 FLUX.1-dev模型,点击下图区域的获取权限,当然我这里已经获取了权限,因此这里提示已经有权限了。这里是重要的设置,若不获取权限,我们的代码是无法进行模型预训练权重的下载的。

获取模型权重

第二步是设置自己的 access token,这里需要点击右上角的头像,选择 access token 就可以了

进入 token 设置界面,我们选择创建新的 token 即可。

在 token 设置界面,设置 token 的名称,然后下面这几个配置也需要勾选上,主要是配置这个 token 的权限,都勾选上,肯定没有错。

设置完成后,就可以点击最下方的按键,确认即可,然后系统会生成一个 token,这里 copy 出来即可。

有了上面的 token,我们就可以继续我们的代码了。

pip install -U diffusers!huggingface-cli login

首先,我们需要安装diffusers库,Hugging Face 的 diffusers 库是一个用于生成扩散模型(Diffusion Models)的 Python 库。扩散模型是一类生成模型,近年来在图像生成、音频生成等领域取得了显著进展。diffusers 库旨在为用户提供一个简单、灵活且高效的接口,以便于使用和实验各种扩散模型。

flux 文生图大模型

然后使用登录一下 hugging face,运行huggingface-cli login,系统会提示需要输入 token,还记得我们 copy 出的 access token 吗,就是这里需要。

输入 token 后,系统提示token 有效,并打印出了 token 的名称,跟我们在 hugging face 官网建立的名称一样,说明我们已经有相关权限来访问我们的模型了。

Token is valid (permission: fineGrained).The token `FLUX` has been saved to /root/.cache/huggingface/stored_tokensToken has not been saved to git credential helper.Your token has been saved to /root/.cache/huggingface/tokenLogin successful.The current active token is: `FLUX`import torchfrom diffusers import FluxPipelinepipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)pipe.enable_model_cpu_offload #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU powerprompt = "A cat holding a sign that says hello world"image = pipe( prompt, height=200, width=200, guidance_scale=3.5, num_inference_steps=50, max_sequence_length=512, generator=torch.Generator("cpu").manual_seed(0)).images[0]image.save("flux-dev.png")

flux 文生图大模型

通过以上的操作,我们再运行代码,就可以正常的下载相关的预训练权重了。然后模型下载完成后,会自动进行代码的运行操作,但是此模型需要至少 20G 的 GPU,若配置较低,可以选择其他低版本的预训练模型权重。

当然 hugging face 上面也提供了试用链接,我们只需要点击进入各个模型,就可以在线体验 flux.1模型,体验了一下,几乎是 几秒钟的时间,系统就会出图,其出图质量与速度还是很惊人的,毕竟是此模型也算是一个比较强大的模型了。而自己电脑运行,其速度极慢,且需要的配置也是偏高,果真大模型也只能这些科技大公司才能玩的转。

flux 文生图大模型

flux 文生图大模型

来源:人工智能研究所

相关推荐