bytez/gustavosta_magicprompt-stable-diffusion

Gustavosta/MagicPrompt-Stable-Diffusion 镜像说明

镜像概述

本Docker镜像封装了Gustavosta/MagicPrompt-Stable-Diffusion文本生成模型，专注于为Stable Diffusion生成高质量提示词（prompt）。该模型支持本地部署，可通过CPU或GPU运行，并提供灵活的参数配置以满足不同生成需求。

核心功能

1. 提示词生成: 根据输入文本生成符合Stable Diffusion要求的创意提示词。
2. 多设备支持: 自动适配CPU/GPU，或手动指定运行设备（cuda/cpu/auto）。
3. 权重缓存: 将模型权重保存到本地目录，避免重复下载，提升部署效率。
4. 参数定制: 支持调整生成长度、温度、top_p等参数，灵活控制生成效果。
5. 流式响应: 支持流式输出生成结果，适用于实时交互场景。

使用场景

• 本地AI创作: 为Stable Diffusion生成个性化提示词，辅助图像生成。
• 工作流集成: 嵌入自动化管道，批量生成提示词。
• 模型测试: 快速验证模型功能与参数调优。

配置说明

前置条件

1. 已安装Docker环境。
2. 获取免费Bytez API密钥（访问Bytez注册登录后获取）。

部署步骤

1. 拉取镜像

bash
docker pull docker.xuanyuan.run/bytez/gustavosta_magicprompt-stable-diffusion

2. 启动容器（默认端口8000）

bash
docker run -it \
 -e KEY=你的_BYTEZ_API_KEY \
 -e PORT=8000 \
 -p 8000:8000 \
 docker.xuanyuan.run/bytez/gustavosta_magicprompt-stable-diffusion

3. 调整端口（示例：80端口）

bash
docker run -it \
 -e KEY=你的_BYTEZ_API_KEY \
 -e PORT=80 \
 -p 80:80 \
 docker.xuanyuan.run/bytez/gustavosta_magicprompt-stable-diffusion

4. 本地权重缓存（推荐）

bash
docker run -it \
 -v /本地缓存目录:/server/model \
 -e HF_HOME=/server/model \
 -e KEY=你的_BYTEZ_API_KEY \
 -p 8000:8000 \
 -e PORT=8000 \
 docker.xuanyuan.run/bytez/gustavosta_magicprompt-stable-diffusion

5. GPU加速运行

bash
docker run -it \
 --gpus all \
 -e KEY=你的_BYTEZ_API_KEY \
 -p 8000:8000 \
 -e PORT=8000 \
 docker.xuanyuan.run/bytez/gustavosta_magicprompt-stable-diffusion

6. GPU+本地缓存

bash
docker run -it \
 --gpus all \
 -v /本地缓存目录:/server/model \
 -e HF_HOME=/server/model \
 -e KEY=你的_BYTEZ_API_KEY \
 -p 8000:8000 \
 -e PORT=8000 \
 docker.xuanyuan.run/bytez/gustavosta_magicprompt-stable-diffusion

调用示例

发送POST请求生成提示词：

bash
curl --location 'http://0.0.0.0:8000/run' \
--header 'Content-Type: application/json' \
--data-raw '{
  "text": "Once upon a time there was a ",
  "stream": false,
  "params": {
    "max_new_tokens": 50,
    "temperature": 1.3,
    "top_p": 1.2
  }
}'

设置"stream": true可获取流式响应。

环境变量说明

• KEY: Bytez API密钥（必填）。
• PORT: 容器监听端口（默认8000）。
• DEVICE: 运行设备（auto/cuda/cpu，默认auto）。
• HF_HOME: 权重缓存目录（配合-v挂载使用）。

参数参考

完整参数列表（如长度控制、生成策略、概率调整等）可查看https://github.com/huggingface/transformers/blob/v4.38.2/src/transformers/generation/configuration_utils.py。

支持与帮助

加入Bytez 社区或发送邮件至获取帮助。

本地部署模型可提升隐私性与控制力，助力AI创作！