VertexAIImageGenerator

Google Vertex AI의 생성 모델로 이미지를 생성해 주는 컴포넌트예요. 텍스트 프롬프트를 주면 그에 맞는 이미지를 만들어 줘요.

필수 run 변수: prompt — 모델에 보낼 프롬프트가 담긴 문자열 출력 변수: images — 모델이 생성한 이미지를 담은 ByteStream 리스트 API 레퍼런스: Google Vertex GitHub 링크: https://github.com/deepset-ai/haystack-core-integrations/tree/main/integrations/google_vertex 패키지 이름: google-vertex-haystack

VertexAIImageGenerator는 imagegeneration 모델을 지원해요.

출처: 문서

본문

파라미터 개요 (Parameters Overview)

VertexAIImageGenerator는 인증에 Google Cloud Application Default Credentials (ADC)를 사용해요. ADC 설정 방법은 공식 문서를 참고하세요.

Google Vertex AI 엔드포인트를 사용하도록 승인된 프로젝트에 접근 권한이 있는 계정을 쓰는 게 중요하다는 점을 기억하세요.

프로젝트 ID는 GCP 리소스 매니저에서 찾거나, 터미널에서 gcloud projects list를 실행해 찾을 수 있어요. gcloud CLI에 대한 자세한 내용은 공식 문서를 참고하세요.

사용법 (Usage)

VertexAIImageGenerator를 쓰려면 google-vertex-haystack 패키지를 설치해야 해요:

pip install google-vertex-haystack

단독으로 사용하기 (On its own)

기본 사용법:

from pathlib import Path
from haystack_integrations.components.generators.google_vertex import (
    VertexAIImageGenerator,
)

generator = VertexAIImageGenerator()
result = generator.run(prompt="Generate an image of a cute cat")
result["images"][0].to_file(Path("my_image.png"))

생성할 이미지 개수나 guidance scale(프롬프트 강도) 같은 다른 파라미터도 설정할 수 있어요.

부정 프롬프트(negative prompt)를 사용해 이미지에서 특정 요소를 빼게 할 수도 있어요:

from pathlib import Path
from haystack_integrations.components.generators.google_vertex import (
    VertexAIImageGenerator,
)

generator = VertexAIImageGenerator(
    number_of_images=3,
    guidance_scale=12,
)
result = generator.run(
    prompt="Generate an image of a cute cat",
    negative_prompt="window, chair",
)
for i, image in enumerate(result["images"]):
    images.to_file(Path(f"image_{i}.png"))

VertexAIImageGenerator