metadata

license: other
license_name: hippojabe1.0
license_link: LICENSE
base_model: runwayml/stable-diffusion-v1-5
pipeline_tag: text-to-image
tags:
  - art
  - hippo

from diffusers import DiffusionPipeline import torch

加載模型

pipeline = DiffusionPipeline.from_pretrained("izzysde/hippojabe_style_image_generator")

如果有 GPU，將模型移到 CUDA

if torch.cuda.is_available(): pipeline.to("cuda") else: pipeline.to("cpu")

定義生成圖像的參數

prompt = "show a cute bird in the style of hippojabe" # 用英文寫描述一定要用in the style of hippojabe結尾 num_inference_steps = 100 # 控制生成質量和速度的步驟數 guidance_scale = 9 # 控制生成圖像與文本提示的一致性

生成圖片

with torch.no_grad(): image = pipeline(prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale)["images"][0]

保存或顯示圖像

image.save("generated_image.png") image.show()