File size: 1,181 Bytes
f7edcff 2f51866 4407962 b302b2d 2f51866 b302b2d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
---
license: other
license_name: hippojabe1.0
license_link: LICENSE
base_model: runwayml/stable-diffusion-v1-5
pipeline_tag: text-to-image
tags:
- art
- hippo
---
使用hippojabe的風格生成卡通畫風圖片
pip install diffusers, accelerate
pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2+cu118 -f https://download.pytorch.org/whl/cu118/torch_stable.html
```python
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()
``` |