部署Qwen-Image
- 安装准备
pip install git+https://github.com/huggingface/diffusers
pip install peft==0.17.0
-
报错：AttributeError: ‘dict’ object has no attribute ‘to_dict’
解决办法：
修改
/opt/conda/lib/python3.11/site-packages/transformers/generation/configuration_utils.py的第1290行，改为:decoder_config_dict = decoder_config -
最终代码
from diffusers import DiffusionPipeline
from tools.prompt_utils import rewrite
import torch# Initialize the pipeline
pipe = DiffusionPipeline.from_pretrained("/root/private_data/models/Qwen/Qwen-Image", torch_dtype=torch.bfloat16)
pipe = pipe.to("cuda")# Generate with different aspect ratios
aspect_ratios = {"16:9": (1664, 960),
}prompt = "帮我画一个农田里面堆肥的场景" # Chinese prompt
#prompt = rewrite(prompt)width, height = aspect_ratios["16:9"]image = pipe(prompt=prompt,width=width,height=height,num_inference_steps=50,true_cfg_scale=4.0,generator=torch.Generator(device="cuda").manual_seed(42)
).images[0]image.save("example.png")