apinaioダウンロード - apinaioソースコードのダウンロード

apinaio

AI ソースコード

1.0.0

ダウンロード

APInaio (一口です)

警告
重い展開中。助けが必要です!

Ollama + ComfyUI + ffmpeg Docker イメージを使用した text2img、img2img、text2video img2video パイプラインでの迅速な開発のための Docker 環境をすぐに使用できる

特徴

多数の LLM が利用可能です。
ComfyUI パッケージの使用と更新が簡単
GPU 対応の ffmpeg により、すべてのエンコード/デコードのニーズに対応します。

設定する

モデルをダウンロードし、API 経由で提供するための./config/ollama/entrypoint.sh
カスタムモデルを定義するための./config/ollama/modelfiles
comfyui とその API をプロビジョニングするための./config/comfyui/provisioning.sh
./config/comfyui/models.csv 、カスタムモデルをダウンロード/更新するために上記で使用されます。 SDXL および AnimateDiff モデルが含まれているため、models.csv に名前を変更して使用します。
./config/comfyui/nodes.csvは、カスタムノードのダウンロード/更新のために上記で使用されます。 SDXL および AnimateDiff ノードが含まれているため、nodes.csv に名前を変更して使用します。

クイックスタート

template.env の名前を .env に変更し、コンテナを 1 つずつ起動します (推奨)

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) でテスト済み

何が含まれていますか?

Ollama-API

TODO 更新の説明

APIの例

利用可能なすべてのモデルを確認する

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"
} '

快適なUI

警告
root として実行します。

GUI はポート 8188 で表示されます。私は通常、ワークフローをブラウザ上に直接ドラッグアンドドロップします。ドライブのスペースが不足している場合は、ボリュームのマウントポイントを変更します。

FFmpeg

警告
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