|
--- |
|
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的風格生成卡通畫風圖片 |
|
|
|
# hippojabe_style_image_generator 是基於 runwayml/stable-diffusion-v1-5 模型,用[小河馬實用貼圖--手繪風](https://store.line.me/stickershop/product/27243710/zh-Hant?from=sticker)作為訓練圖像,用 DreamBooth 方法去 fine-tune 的 text to image 模型 |
|
|
|
|
|
## 安裝 diffusers, accelerate, cuda版torch |
|
```bash |
|
pip install diffusers, accelerate, 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() |
|
|
|
``` |