stable-diffusion-3.5-medium图生图怎么调用
时间: 2025-07-05 08:03:07 浏览: 23
### 使用 `stable-diffusion-3.5-medium` 进行图像生成
为了使用 `stable-diffusion-3.5-medium` 模型进行图像生成,需先确保环境配置正确并安装必要的依赖项。完成这些准备工作之后,可以通过 Python 脚本调用该模型。
#### 安装 Git LFS 和克隆仓库
首先,需要通过 Git 复制包含模型文件的存储库至本地机器:
```bash
git lfs install
git clone https://www.modelscope.cn/AI-ModelScope/stable-diffusion-3-medium.git stable-diffusion-3-medium
```
此命令会设置好大型文件支持 (LFS),并将指定版本的 Stable Diffusion 模型下载下来[^1]。
#### 安装依赖包
接着,在项目目录下执行 pip 命令来安装所需的Python 库和其他资源:
```bash
pip install -r requirements.txt
```
这一步骤确保所有必需的软件组件都已就绪以便后续操作能够顺利运行[^2]。
#### 加载预训练模型
一旦上述步骤顺利完成,则可通过如下方式加载预训练好的 `diffusers` 版本的 `stable-diffusion-3.5-medium` 模型实例化对象用于进一步处理:
```python
from diffusers import StableDiffusionPipeline, EulerAncestralDiscreteScheduler
import torch
model_id = "AI-ModelScope/stable-diffusion-3-medium"
scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16).to("cuda")
prompt = "A photo of an astronaut riding a horse on Mars."
image = pipe(prompt).images[0]
image.save("astronaut_rides_horse.png")
```
这段代码展示了如何利用 Hugging Face 的 `transformers` 及其子库 `diffusers` 来快速简便地获取并应用最新的 AI 图像生成功能[^3]。
阅读全文
相关推荐



















