Google Colab에서 stable-diffusion--large-turbo를 실행하는 방법

stable-diffusion-3.5-large-turbo는 고정밀 텍스트-이미지 모델입니다.

이 가이드에서는 Google Colab에서 모델을 설정하고 실행하는 방법을 설명합니다.

---

전제 조건

허깅페이스를 방문해 보세요.

stable-diffusion-3.5-large-turbo를 사용하려면 Huggingface 계정이 필요합니다.

아직 계정이 없다면 계정을 만들어주세요.

회원가입을 마치면 다음 화면이 나타납니다.

필수 정보를 입력하시면 바로 모델에 접속하실 수 있습니다.

모델을 다운로드하여 사용하려면 액세스 토큰이 필요합니다. 계정 페이지에서 계정을 만드세요.

오른쪽 상단에 있는 프로필 아이콘을 통해 계정 페이지로 이동한 후 액세스 토큰 탭으로 이동한 후 새 토큰 만들기를 선택하여 토큰을 만드세요.

---

코드 실행

필수 라이브러리 설치

먼저 Google Colab에 필요한 라이브러리를 설치합니다.

!pip install --quiet -U transformers

-U 옵션은 라이브러리를 최신 버전으로 업데이트하고 --quiet는 다운로드 메시지를 표시하지 않습니다.

계정 인증

다음 명령을 실행하고 이전에 생성한 토큰을 입력하여 Huggingface 계정을 인증하세요.

!huggingface-cli login

모델 다운로드

다음 Python 코드를 사용하여 모델을 로드하고 설정합니다.

import torch
from diffusers import StableDiffusion3Pipeline

pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large-turbo", torch_dtype=torch.bfloat16)
pipe = pipe.to("cuda")

참고: 이 모델은 약 27GB의 메모리를 소비합니다.

---

이미지 생성

다음 코드를 실행하여 이미지를 생성하여 설정을 테스트하세요.

prompt = "A capybara holding a sign that reads Hello Fast World"
save_filename = "capybara.png"
image = pipe(
    prompt,
    num_inference_steps=4,
    guidance_scale=0.0,
).images[0]

Diffusers GitHub 문서에서 이러한 인수에 대한 설명을 찾을 수 있습니다.

생성된 이미지 저장 및 표시:

image.save(save_filename)
image

위 내용은 Google Colab에서 stable-diffusion--large-turbo를 실행하는 방법의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!