dede网站地图位置,音乐网站开发与需求,企业网站开源系统,那些行业做网站优化的比较多写在最前#xff0c;参加DataWhale 10月组队学习。 参考资料#xff1a; HuggingFace 开源diffusion-models-class
1.扩散模型介绍
2.调用模型生成一张赛博风格的猫咪图片
2.1 安装依赖包
%pip install -qq -U diffusers datasets transformers accelerate ftfy pyarrow9…写在最前参加DataWhale 10月组队学习。 参考资料 HuggingFace 开源diffusion-models-class
1.扩散模型介绍
2.调用模型生成一张赛博风格的猫咪图片
2.1 安装依赖包
%pip install -qq -U diffusers datasets transformers accelerate ftfy pyarrow9.0.02.2 notebook登录huggingface账号
from huggingface_hub import notebook_loginnotebook_login()建议创建可写token后续如果要上传模型的话有write权限 https://huggingface.co/settings/tokens
2.3 配置命令
%%capture
!sudo apt -qq install git-lfs
!git config --global credential.helper store这两个命令用于在Jupyter Notebook中运行Shell命令并将输出捕获capture起来而不直接在Notebook中显示输出。
%%capture是一个Jupyter Notebook的魔法命令magic command它会捕获后续单元格中所有命令的标准输出stdout和标准错误输出stderr并将其保存在一个变量中以便进一步处理或分析。
!sudo apt -qq install git-lfs用于在Shell中以sudo权限管理员权限安装git-lfsGit Large File Storage。-qq参数表示安装过程中不显示详细信息。
!git config --global credential.helper store用于在Shell中设置git的全局配置将认证信息存储在文件中而不是在内存中。这可以使得git在每次需要认证时都不再询问用户名和密码而是从文件中读取认证信息。
请注意sudo命令需要输入密码来确认你有权限进行系统级的安装。在Jupyter Notebook中执行这些命令时你可能需要手动输入密码。如果你使用的是虚拟环境你可能不需要使用sudo权限来安装包。请根据你的需求和系统权限进行相应的调整。
2.4 引入依赖
import numpy as np
import torch
import torch.nn.functional as F
from matplotlib import pyplot as plt
from PIL import Image
# Mac users may need device mps (untested)
device torch.device(cuda if torch.cuda.is_available() else cpu)2.5 下载模型
from diffusers import StableDiffusionPipeline
import torch
model_id prompthero/openjourney
pipe StableDiffusionPipeline.from_pretrained(model_id, torch_dtypetorch.float16)
pipe pipe.to(cuda)2.6 输入提示词出图
prompt A cat wearing smart sunglasses, holding a laser, wearing a space suit, cyberpunk style
image pipe(prompt).images[0]
image.save(./cat1.png)
image
