izzysde commited on
Commit
f7edcff
1 Parent(s): 568f1e4

update .md

Browse files

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()

Files changed (1) hide show
  1. README.md +10 -5
README.md CHANGED
@@ -1,5 +1,10 @@
1
- ---
2
- license: other
3
- license_name: hippojabe1.0
4
- license_link: LICENSE
5
- ---
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ license_name: hippojabe1.0
4
+ license_link: LICENSE
5
+ base_model: runwayml/stable-diffusion-v1-5
6
+ pipeline_tag: text-to-image
7
+ tags:
8
+ - art
9
+ - hippo
10
+ ---