ZacLiu commited on
Commit
12cba2c
·
1 Parent(s): 0e650ca

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -0
README.md CHANGED
@@ -1,3 +1,17 @@
1
  ---
2
  license: creativeml-openrail-m
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: creativeml-openrail-m
3
  ---
4
+ #This is a DreamBooth model trained on several public available images from web.
5
+ ```
6
+ from diffusers import AltDiffusionPipeline, DPMSolverMultistepScheduler
7
+ import torch
8
+
9
+ pipe = AltDiffusionPipeline.from_pretrained("BAAI/DreamBooth-AltDiffusion")
10
+ pipe = pipe.to("cuda")
11
+
12
+ pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
13
+
14
+ prompt = "一张<鸣人>男孩的照片"
15
+ image = pipe(prompt, num_inference_steps=25).images[0]
16
+ image.show()
17
+ ```