Unduh apinaio - pengunduhan kode sumber apinaio

apinaio

Kode Sumber AI

1.0.0

Unduh

APInAIo (Ini seteguk)

Peringatan
Dalam perkembangan yang berat. Bantuan diperlukan!

Lingkungan buruh pelabuhan siap digunakan untuk pengembangan cepat dalam pipeline text2img, img2img, text2video img2video menggunakan gambar Docker Ollama + ComfyUI + ffmpeg

Fitur

Banyak LLM yang tersedia.
Mudah digunakan dan memperbarui paket ComfyUI
ffmpeg yang diaktifkan GPU untuk semua kebutuhan pengkodean/dekode.

Konfigurasikan

./config/ollama/entrypoint.sh untuk mengunduh model dan menyajikannya melalui API
./config/ollama/modelfiles untuk mendefinisikan model khusus
./config/comfyui/provisioning.sh untuk menyediakan comfortui dan API-nya
./config/comfyui/models.csv digunakan di atas untuk mengunduh/memperbarui model khusus. Model SDXL dan AnimateDiff disertakan, ganti namanya menjadi models.csv untuk digunakan.
./config/comfyui/nodes.csv digunakan di atas untuk mengunduh/memperbarui node khusus. Node SDXL dan AnimateDiff disertakan, ganti namanya menjadi node.csv untuk digunakan.

Mulai cepat

Ganti nama template.env menjadi .env lalu luncurkan container satu per satu (disarankan)

docker-compose up ollama-api -d
docker-compose up comfyui-api -d
docker-compose up ffmpeg-api -d

Setelah mereka berjalan, Anda dapat menjatuhkan semuanya

docker compose down

Dan bawa semuanya saat dibutuhkan

docker compose up -d

Saat diuji pada Python 3.10 / Pytorch 2.0.1, NVIDIA CUDA (WSL)

Apa yang Termasuk?

Ollama-API

Deskripsi Pembaruan TODO

Contoh API

Periksa semua model yang tersedia

curl http://localhost:11434/api/tags

Gunakan model generator prompt tengah perjalanan

curl -X POST http://localhost:11434/api/generate -d ' {
  "model": "mj:latest",
  "prompt": "a sports car in the mountains.",
  "stream": false} '

Lihat informasi model

http://localhost:11434/api/show -d ' {
  "name": "mj:latest"
} '

UI yang nyaman

Peringatan
Berjalan sebagai root.

GUI terlihat di port 8188. Saya biasanya drag dan drop alur kerja saya langsung di browser. Jika ruang Drive terbatas, ubah titik pemasangan volume.

FFmpeg

Peringatan
Berjalan sebagai root.

Deskripsi Pembaruan TODO

Contoh penggunaan

Konversi mp4 ke 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

Ubah png menjadi 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 '

Konversi gambar ke video 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