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介面 |