Amazon SageMaker AI에서 vLLM-Omni를 사용한 이미지 및 비디오 생성
Generate images and video with vLLM-Omni on SageMaker AI – Part 2
이 글에서는 Amazon SageMaker AI에서 텍스트 프롬프트를 이미지로 변환하고, 이를 애니메이션 비디오로 변환하는 방법을 설명한다. AWS vLLM-Omni Deep Learning Container(DLC)를 사용하여 FLUX.2-klein-4B 이미지 생성과 Wan2.1-VACE-1.3B 비디오 생성을 위한 두 개의 엔드포인트를 배포한다. 이 워크플로우는 텍스트 프롬프트를 통해 정적 이미지를 생성하고, 생성된 이미지를 비디오 엔드포인트로 전달하여 MP4 파일을 Amazon S3에서 가져온다.
FLUX.2-klein-4B와 Wan2.1-VACE-1.3B 모델을 통해 이미지와 비디오 생성을 분리하여 각 모델에 적합한 인스턴스 유형과 추론 옵션을 사용할 수 있다.
원문 출처
AWS Machine Learning Blog