
一、下载模型
bash
pip install modelscope
modelscope download --model AI-ModelScope/stable-diffusion-xl-base-1.0
注意模型下载路径

之后可以把它移动到固定的位置
二、下载依赖项
现在来到模型文件夹的同级目录

创建虚拟环境3.10
bash
conda create -n 你得环境 python=3.10
安装:
bash
apt-get update
apt-get install libgl1-mesa-glx
pip install invisible_watermark transformers safetensors
pip install diffusers==0.18.0
pip install addict
pip install datasets==2.21.0
pip install simplejson
pip install sortedcontainers
pip install torchvision
pip install "setuptools<81"
三、使用模型
创建 .py 文件
bash
from modelscope.utils.constant import Tasks
from modelscope.pipelines import pipeline
import cv2
pipe = pipeline(task=Tasks.text_to_image_synthesis,
model='AI-ModelScope/stable-diffusion-xl-base-1.0',
use_safetensors=True,
model_revision='v1.0.0')
prompt = 'The princess kisses the prince, and they held a wedding, Realistic portrait'
output = pipe({
'text': prompt,
'height': 576, # 指定高度
'width': 1024 # 指定宽度
})
cv2.imwrite('result.png', output['output_imgs'][0])
四、报错处理
如果这几个出现报错,直接把他们卸载
bash
pip uninstall diffusers huggingface_hub transformers
再重新一键安装
bash
pip install diffusers huggingface_hub transformers

五、风格词
电影
自然主义
动态摄影
浪漫主义
古典