apinaio
1.0.0
警告
正在大力发展中。需要帮助!
准备使用 Docker 环境,使用 Ollama + ComfyUI + ffmpeg Docker 镜像在 text2img、img2img、text2video img2video 管道中进行快速开发
./config/ollama/entrypoint.sh
用于下载模型并通过 API 提供服务./config/ollama/modelfiles
用于定义自定义模型./config/comfyui/provisioning.sh
用于配置 comfyui 及其 API./config/comfyui/models.csv
上面用于下载/更新自定义模型。包含SDXL和AnimateDiff模型,重命名为models.csv以供使用。./config/comfyui/nodes.csv
上面用于下载/更新自定义节点。包含SDXL和AnimateDiff节点,重命名为nodes.csv以供使用。 将 template.env 重命名为 .env,然后一一启动容器(推荐)
docker-compose up ollama-api -d
docker-compose up comfyui-api -d
docker-compose up ffmpeg-api -d
一旦它们开始运行,你就可以将它们全部关闭
docker compose down
并在需要时调出所有内容
docker compose up -d
在 Python 3.10 / Pytorch 2.0.1、NVIDIA CUDA (WSL) 上测试
TODO 更新说明
检查所有可用型号
curl http://localhost:11434/api/tags
使用中途提示生成器模型
curl -X POST http://localhost:11434/api/generate -d ' {
"model": "mj:latest",
"prompt": "a sports car in the mountains.",
"stream": false} '
查看型号信息
http://localhost:11434/api/show -d ' {
"name": "mj:latest"
} '
警告
以 root 身份运行。
GUI 在端口 8188 上可见。我通常将工作流程直接拖放到浏览器上。如果驱动器空间不足,请更改卷安装点。
警告
以 root 身份运行。
TODO 更新说明
将 mp4 转换为 png
docker exec -it ffmpeg-api ffmpeg -y -hwaccel cuvid -c:v h264_cuvid -resize 576x1024 -i /output/Dancing.mp4 -vf " scale_npp=format=yuv420p,hwdownload,format=yuv420p " -pix_fmt yuvj420p -color_range 2 /output/frame_%03d.jpg
将 png 转换为 jpg
docker exec -it ffmpeg-api /bin/bash -c ' for image in /output/*.png; do ffmpeg -i "$image" "${image%.png}.jpg"; rm "$image"; echo "image $image converted to ${image%.png}.jpg "; done '
将图像转换为 mp4 视频
docker exec -it ffmpeg-api ffmpeg -y -loglevel error -i ' /output/frame_%03d.jpg ' -r 30 -c:v hevc_nvenc -pix_fmt yuv420p -preset fast /output/final.mp4
需要公开某些端口才能使服务运行或所提供软件的某些功能正常运行
开放端口 | 服务/说明 |
---|---|
11434 | 奥拉马网络服务器 |
8188 | 舒适UI界面 |