Skip to content

Commit 7f00a08

Browse files
authored
Update README.md
1 parent e4040b1 commit 7f00a08

1 file changed

Lines changed: 81 additions & 0 deletions

File tree

‎projects/blip-diffusion/README.md‎

Lines changed: 81 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,87 @@ pip install -e .
3131
- **Virtual Try-On via Subject-driven Editing**:
3232
- the model can be used to naturally facilitate virtual try-on. We provide an zero-shot example: [notebook](https://github.com/salesforce/LAVIS/blob/main/projects/blip-diffusion/notebooks/editing_tryon_zeroshot.ipynb), [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/salesforce/LAVIS/blob/main/projects/blip-diffusion/notebooks/editing_tryon_zeroshot.ipynb);
3333

34+
### **🧨 Diffusers Support**
35+
BLIP-Diffusion is now supported in 🧨[Diffusers](https://huggingface.co/docs/diffusers/main/en/api/pipelines/blip_diffusion).
36+
- Example on subject-driven generation:
37+
```python
38+
from diffusers.pipelines import BlipDiffusionPipeline
39+
from diffusers.utils import load_image
40+
import torch
41+
42+
blip_diffusion_pipe = BlipDiffusionPipeline.from_pretrained(
43+
"Salesforce/blipdiffusion", torch_dtype=torch.float16
44+
).to("cuda")
45+
46+
47+
cond_subject = "dog"
48+
tgt_subject = "dog"
49+
text_prompt_input = "swimming underwater"
50+
51+
cond_image = load_image(
52+
"https://huggingface.co/datasets/ayushtues/blipdiffusion_images/resolve/main/dog.jpg"
53+
)
54+
guidance_scale = 7.5
55+
num_inference_steps = 25
56+
negative_prompt = "over-exposure, under-exposure, saturated, duplicate, out of frame, lowres, cropped, worst quality, low quality, jpeg artifacts, morbid, mutilated, out of frame, ugly, bad anatomy, bad proportions, deformed, blurry, duplicate"
57+
58+
59+
output = blip_diffusion_pipe(
60+
text_prompt_input,
61+
cond_image,
62+
cond_subject,
63+
tgt_subject,
64+
guidance_scale=guidance_scale,
65+
num_inference_steps=num_inference_steps,
66+
neg_prompt=negative_prompt,
67+
height=512,
68+
width=512,
69+
).images
70+
output[0].save("image.png")
71+
```
72+
- Example on subject-driven stylization
73+
```python
74+
from diffusers.pipelines import BlipDiffusionControlNetPipeline
75+
from diffusers.utils import load_image
76+
from controlnet_aux import CannyDetector
77+
import torch
78+
79+
blip_diffusion_pipe = BlipDiffusionControlNetPipeline.from_pretrained(
80+
"Salesforce/blipdiffusion-controlnet", torch_dtype=torch.float16
81+
).to("cuda")
82+
83+
style_subject = "flower"
84+
tgt_subject = "teapot"
85+
text_prompt = "on a marble table"
86+
87+
cldm_cond_image = load_image(
88+
"https://huggingface.co/datasets/ayushtues/blipdiffusion_images/resolve/main/kettle.jpg"
89+
).resize((512, 512))
90+
canny = CannyDetector()
91+
cldm_cond_image = canny(cldm_cond_image, 30, 70, output_type="pil")
92+
style_image = load_image(
93+
"https://huggingface.co/datasets/ayushtues/blipdiffusion_images/resolve/main/flower.jpg"
94+
)
95+
guidance_scale = 7.5
96+
num_inference_steps = 50
97+
negative_prompt = "over-exposure, under-exposure, saturated, duplicate, out of frame, lowres, cropped, worst quality, low quality, jpeg artifacts, morbid, mutilated, out of frame, ugly, bad anatomy, bad proportions, deformed, blurry, duplicate"
98+
99+
100+
output = blip_diffusion_pipe(
101+
text_prompt,
102+
style_image,
103+
cldm_cond_image,
104+
style_subject,
105+
tgt_subject,
106+
guidance_scale=guidance_scale,
107+
num_inference_steps=num_inference_steps,
108+
neg_prompt=negative_prompt,
109+
height=512,
110+
width=512,
111+
).images
112+
output[0].save("image.png")
113+
```
114+
34115

35116
### Cite BLIP-Diffusion
36117
If you find our work helpful, please consider citing:

0 commit comments

Comments
 (0)