github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-05 02:51:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

[example] fix save_load bug for dreambooth (#2280)

Browse Source
This commit is contained in:
BlueRum
2023-01-03 17:13:29 +08:00
committed by GitHub
parent f027ef7913
commit 1405b4381e
5 changed files with 53 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
examples/images/dreambooth/inference.py Normal file
View File

@@ -0,0 +1,12 @@
from diffusers import StableDiffusionPipeline, DiffusionPipeline
import torch
model_id = <Your Model Path>
print(f"Loading model... from{model_id}")
pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to("cuda")
prompt = "A photo of an apple."
image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5).images[0]
image.save("output.png")