shortgiraffe/runpod-image Docker 镜像 - 轩辕镜像

Automatic1111 | Worker

这是一个RunPod worker，基于Stable Diffusion WebUI（Stable Diffusion AI模型的用户界面），使用Stable Diffusion模型执行AI任务。该worker针对RunPod环境进行了优化，提供API接口支持推理请求处理。

模型

worker使用针对RunPod优化的Stable Diffusion模型，以SafeTensors格式存储（一种便于AI模型高效加载和执行的格式）。可从指定链接下载模型文件。

构建Worker

通过Dockerfile构建worker，Dockerfile内容包括：

• 指定基础镜像、环境变量
• 系统包依赖和Python依赖（Python依赖在requirements.txt中，主要依赖为runpod==0.9.4）
• 安装和设置Stable Diffusion WebUI的步骤
• 下载模型文件
• 使用supervisor设置API服务器

运行Worker

通过start.sh脚本运行worker，该脚本启动init系统并运行serverless handler脚本。

API

worker提供推理API，由supervisor配置（配置文件为webui_api.conf），API运行在端口3000。

Serverless Handler

serverless handler（rp_handler.py）是处理推理请求的Python脚本，定义了handler(event)函数：接收推理请求，使用Stable Diffusion模型执行推理，返回输出结果。