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 노드가 포함되어 있으며 사용하려면 node.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)에서 테스트되었습니다.

무엇이 포함되어 있나요?

올라마-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"
} '

ComfyUI

경고
루트로 실행됩니다.

GUI는 포트 8188에서 볼 수 있습니다. 저는 일반적으로 작업 흐름을 브라우저에 직접 끌어다 놓습니다. 드라이브 공간이 부족한 경우 볼륨 장착 지점을 변경하십시오.

FFmpeg

경고
루트로 실행됩니다.

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