코랩에서 스테이블 디퓨전 사용해 보기 colab stable diffusion text2image

코랩에서 스테이블 디퓨전 사용해 보기 - colab stable diffusion, text2image¶

• 아래 그림과 같이 매뉴에서 런타임 클릭 후 T4 GPU 사용을 설정해주세요

In [1]:

# Colab에 필요한 라이브러리 설치하기.
!pip install --upgrade diffusers accelerate transformers

In [2]:

# 모델 불러오기
from diffusers import DiffusionPipeline
pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")

In [ ]:

## gpu에 모델 올리기
pipeline.to("cuda")

Out[ ]:

StableDiffusionPipeline {
  "_class_name": "StableDiffusionPipeline",
  "_diffusers_version": "0.30.3",
  "_name_or_path": "runwayml/stable-diffusion-v1-5",
  "feature_extractor": [
    "transformers",
    "CLIPImageProcessor"
  ],
  "image_encoder": [
    null,
    null
  ],
  "requires_safety_checker": true,
  "safety_checker": [
    "stable_diffusion",
    "StableDiffusionSafetyChecker"
  ],
  "scheduler": [
    "diffusers",
    "PNDMScheduler"
  ],
  "text_encoder": [
    "transformers",
    "CLIPTextModel"
  ],
  "tokenizer": [
    "transformers",
    "CLIPTokenizer"
  ],
  "unet": [
    "diffusers",
    "UNet2DConditionModel"
  ],
  "vae": [
    "diffusers",
    "AutoencoderKL"
  ]
}

In [ ]:

# 이미지 생성을 위한 텍스트 작성 및 이미지 생성
image = pipeline("a image of dog in picasso style").images[0]
image

Out[ ]:

In [ ]:

# 생성된 이미지 저장
image.save("image01.png")

• 생성된 이미지를 코랩에서 다운로드 내 컴퓨터로 다운 받는 방법 입니다.
• 좌측 메뉴바를 이용하여 다운로드 받아 주세요

거실 및 부부화장실 치약짜개 1set(2개)

1set 구매하셔서 안방 및 거실 화장실 모두에서 사용하세요