PromptEnhancer

环境要求

• Python 环境与 pip（README 使用 pip install 安装依赖）
• 能访问 GitHub 与 Hugging Face 下载代码和模型权重
• 使用 GGUF 量化推理时需要 CUDA 支持的显卡环境
• 本地磁盘空间用于存放 7B/32B 模型文件

安装与启动步骤

1. 1获取项目代码

   从官方仓库克隆代码到本地，后续所有命令都在项目根目录执行。

   git clone https://github.com/Hunyuan-PromptEnhancer/PromptEnhancer
   cd PromptEnhancer

   复制

2. 2安装标准依赖

   README 推荐的标准安装方式，按 requirements.txt 一次性装好全部 Python 依赖。

   pip install -r requirements.txt

   复制

3. 3可选：GGUF 安装

   需要更低显存、更快推理（尤其 32B 模型）时改用此方式，脚本会自动加执行权限并安装。

   chmod +x script/install_gguf.sh && ./script/install_gguf.sh

   复制

4. 4准备模型文件

   把下载的模型权重放到项目 models/ 目录，推理脚本会自动检测该目录下的模型。README 未给出下载命令，请自行从官方模型页获取。

5. 5运行命令行推理

   执行 GGUF 命令行脚本，自动读取 models/ 目录模型，输入提示词后得到重写后的结构化提示词。

   python inference/prompt_enhancer_gguf.py

   复制

如何确认成功

运行 inference/prompt_enhancer_gguf.py 后终端正常输出重写后的提示词，且未报找不到模型或依赖缺失的错误。

常见问题

注意事项

• GGUF 安装方式需要 CUDA 支持，纯 CPU 环境请选标准安装。
• README 未提供模型权重的下载命令，需自行从 Hugging Face 官方页面获取后放入 models/ 目录。
• 图像编辑/图生图任务只支持 32B 模型，7B 仅适用于纯文本生图提示词增强。