在线时间621 小时
UID3436040
注册时间2018-5-6
NXP金币6135
TA的每日心情 | 开心 昨天 09:28 |
|---|
签到天数: 1374 天 连续签到: 13 天 [LV.10]以坛为家III
金牌会员
 
- 积分
- 19103
- 最后登录
- 2026-4-7
|
发表于 2026-2-9 23:54:42
|
显示全部楼层
回帖奖励 +2 NXP金币
我也来发一个:
使用sd,模型是dreamlike-diffusion-1.0
from diffusers import StableDiffusionPipeline
import torch
# 加载模型
model_id = "../models/dreamlike-diffusion-1.0"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
device = "cuda" if torch.cuda.is_available() else "cpu"
pipe = pipe.to(device)
prompt = "A majestic horse running in a sunset field, photorealistic style"
negative_prompt = "blurry, low quality, cartoon" # 排除不想要的风格
# 生成图片(默认尺寸512x512)
image = pipe(
prompt=prompt,
negative_prompt=negative_prompt,
num_inference_steps=50, # 生成步数(影响细节)
guidance_scale=7.5, # 文本相关性(值越高越贴近描述)
).images[0]
# 保存图片
image.save("horse_sunset.png")
print("图片已保存")
|
|