首页
/ OneButtonPrompt 使用指南:从安装到生成的完整流程

OneButtonPrompt 使用指南:从安装到生成的完整流程

2026-02-06 05:14:18作者:郦嵘贵Just

概述

OneButtonPrompt 是一个专为 Stable Diffusion 设计的 AI 提示词生成工具,旨在通过自动化流程简化高质量图像生成过程。本教程将从项目结构解析开始,逐步引导您完成环境配置、启动流程及核心功能使用,帮助您快速掌握这款工具的全部潜力。

项目结构解析

核心目录组织

OneButtonPrompt 采用模块化设计,主要目录结构如下:

OneButtonPrompt/
├── 核心脚本/           # 主功能实现
├── csvfiles/           # 提示词组件数据库
├── automated_outputs/  # 自动生成文件存储
├── userfiles/          # 用户自定义资源
├── scripts/            # API 及扩展脚本
└── user_guides/        # 使用文档

关键文件功能

  • 主程序入口main.py 提供图像生成的完整工作流控制
  • 提示词生成build_dynamic_prompt.py 实现核心的动态提示词构建逻辑
  • 预设管理one_button_presets.py 处理用户预设的加载与保存
  • API 接口scripts/api.py 提供外部调用的接口服务

核心功能模块

通过分析代码结构,可识别出以下关键功能模块:

  1. 动态提示词构建build_dynamic_prompt() 函数支持多维度参数控制,生成高度定制化的提示词
  2. 图像生成控制call_txt2img()call_img2img() 分别处理文本转图像和图像转图像功能
  3. 提示词增强enhance_positive()artify_prompt() 提供提示词优化和艺术化处理
  4. 批量处理generateimages() 支持多参数控制的批量生成任务

环境准备与安装

系统要求

  • Python 3.8 或更高版本
  • 符合 Stable Diffusion 运行要求的硬件配置
  • 已安装 Stable Diffusion WebUI 及其 API 服务

安装步骤

  1. 获取项目代码库:

    git clone https://gitcode.com/gh_mirrors/on/OneButtonPrompt
    cd OneButtonPrompt
    
  2. 安装依赖包:

    pip install -r requirements.txt
    
  3. 配置模型(可选):

    python superprompter/download_models.py
    

快速启动指南

基本启动方式

在项目根目录执行以下命令启动基本功能:

python main.py

命令行参数说明

常用参数示例:

# 生成 5 张图像,中等复杂度提示词
python main.py --amount 5 --insanitylevel 5

# 使用特定模型和采样方法
python main.py --model "SDXL 1.0" --samplingmethod "DPM++ 2M Karras"

# 启用高质量修复和放大
python main.py --hiresfix True --upscaler "R-ESRGAN 4x+"

配置文件详解

项目使用 pyproject.toml 管理依赖和项目信息:

[tool.poetry]
name = "OneButtonPrompt"
version = "0.1.0"
description = "A full AI prompt generator for Stable Diffusion"
authors = ["AIrjen <airjen@example.com>"]
license = "GPL-3.0"

[tool.poetry.dependencies]
python = "^3.8"
numpy = "^1.21.0"
# 其他依赖项...

核心功能使用

提示词生成

使用 build_dynamic_prompt.py 生成提示词:

from build_dynamic_prompt import build_dynamic_prompt

prompt = build_dynamic_prompt(
    insanitylevel=7,
    forcesubject="human",
    artists="Van Gogh, Picasso",
    imagetype="oil painting",
    prefixprompt="masterpiece, best quality",
    suffixprompt="intricate details, 8k resolution"
)
print(prompt)

图像生成工作流

  1. 文本转图像基础流程

    from main import generateimages
    
    generateimages(
        amount=3,
        model="SDXL 1.0",
        insanitylevel=6,
        subject="portrait",
        imagetype="photorealistic"
    )
    
  2. 图像转图像高级应用

    from call_img2img import call_img2img
    
    call_img2img(
        imagelocation="input.jpg",
        prompt="remaster, enhance details",
        denoising_strength=0.4,
        scale=2
    )
    

预设管理

  1. 保存自定义预设

    from one_button_presets import OneButtonPresets
    
    presets = OneButtonPresets()
    presets.save_obp_preset({
        "name": "my_portrait_preset",
        "insanitylevel": 7,
        "artists": "Greg Rutkowski, Artgerm",
        "imagetype": "digital art"
    })
    
  2. 加载预设进行生成

    generateimages(
        amount=5,
        OBP_preset="my_portrait_preset",
        subject="female elf"
    )
    

高级功能指南

提示词增强技术

OneButtonPrompt 提供多种提示词增强方法:

  1. 艺术化处理

    from build_dynamic_prompt import artify_prompt
    enhanced_prompt = artify_prompt(prompt="a beautiful landscape", artists="J.M.W. Turner", amountofartists=2)
    
  2. 细节扩充

    from build_dynamic_prompt import flufferizer
    detailed_prompt = flufferizer(prompt="a cat", amountoffluff="high")
    

API 接口使用

通过 scripts/api.py 可以将功能集成到其他应用:

import requests

# 获取随机提示词
response = requests.post(
    "http://localhost:5000/api/random_prompts",
    json={"numberofprompts": 3, "insanitylevel": 5}
)
prompts = response.json()

自定义资源管理

用户可通过 userfiles/ 目录扩展工具功能:

  1. 添加自定义 CSV 提示词组件
  2. 创建个人艺术家风格列表
  3. 定义专属预设模板

常见问题解决

性能优化建议

  • 对于低配置系统,降低 insanitylevel 可减少提示词复杂度
  • 使用 qualitygate 参数控制输出质量与生成速度平衡
  • 预加载常用模型可显著减少生成延迟

错误排查指南

  • API 连接问题:确保 Stable Diffusion WebUI 已启用 API 并检查端口配置
  • 提示词生成失败:检查 CSV 文件格式及完整性,特别是自定义资源
  • 图像质量不佳:调整 cfg 参数(建议 7-10)和采样步数(建议 30+)

总结

OneButtonPrompt 通过模块化设计和灵活的参数控制,为 Stable Diffusion 用户提供了强大的提示词生成和图像创建工具。无论是初学者还是高级用户,都能通过本教程掌握从基础安装到高级功能应用的完整流程。通过合理利用预设系统和自定义资源,可大幅提升 AI 图像生成的效率和质量。

如需进一步学习,可参考 user_guides/ 目录下的专项文档,深入了解各功能模块的详细使用方法。

登录后查看全文
热门项目推荐
相关项目推荐