Update README.md
Browse files
README.md
CHANGED
|
@@ -17,6 +17,21 @@ This pipeline shows how to superimpose different text prompts from [Stable Diffu
|
|
| 17 |
<img src="https://huggingface.co/superdiff/superdiff-sd-v1-4/resolve/main/superdiff_small.gif" alt="drawing" style="width:500px;">
|
| 18 |
</p>
|
| 19 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
|
| 21 |
## Example usage
|
| 22 |
|
|
@@ -33,14 +48,14 @@ image.save("superdiff_output.png")
|
|
| 33 |
|
| 34 |
Arguments that can be set by user in `pipeline()`:
|
| 35 |
|
| 36 |
-
- `prompt_1
|
| 37 |
-
- `prompt_2
|
| 38 |
-
- `seed
|
| 39 |
-
- `num_inference_steps
|
| 40 |
-
- `batch_size
|
| 41 |
-
- `lift
|
| 42 |
-
- `guidance_scale
|
| 43 |
-
- `height`, `width
|
| 44 |
|
| 45 |
To replicate images from Section 4.2 of the paper, you can use the following:
|
| 46 |
|
|
|
|
| 17 |
<img src="https://huggingface.co/superdiff/superdiff-sd-v1-4/resolve/main/superdiff_small.gif" alt="drawing" style="width:500px;">
|
| 18 |
</p>
|
| 19 |
|
| 20 |
+
## Requirements
|
| 21 |
+
|
| 22 |
+
This pipeline can be run with the following packages & versions:
|
| 23 |
+
- PyTorch 2.5.1
|
| 24 |
+
- Diffusers 0.32.1
|
| 25 |
+
- Accelerate 1.2.1
|
| 26 |
+
- Transformers 4.47.1
|
| 27 |
+
|
| 28 |
+
You can install these with:
|
| 29 |
+
|
| 30 |
+
```
|
| 31 |
+
pip install torch
|
| 32 |
+
pip install diffusers accelerate transformers
|
| 33 |
+
```
|
| 34 |
+
|
| 35 |
|
| 36 |
## Example usage
|
| 37 |
|
|
|
|
| 48 |
|
| 49 |
Arguments that can be set by user in `pipeline()`:
|
| 50 |
|
| 51 |
+
- `prompt_1` [required]: text prompt describing first concept to superimpose (e.g. "a flamingo")
|
| 52 |
+
- `prompt_2`[required]: text prompt describing second concept to superimpose (e.g. "a candy cane")
|
| 53 |
+
- `seed`[optional: default=None]: seed for random noise generator for reproducibility; for non-deterministic outputs, set to `None`
|
| 54 |
+
- `num_inference_steps`[optional: default=1000]: number of denoising steps (we recommend 1000!)
|
| 55 |
+
- `batch_size` [optional: default=1]: batch size
|
| 56 |
+
- `lift` [optional: default=0.0]: bias value that favours generation towards one prompt over the other
|
| 57 |
+
- `guidance_scale` [optional: default=7.5]: scale for classifier-free guidance
|
| 58 |
+
- `height`, `width` [optional: default=512]: height and width of generated images
|
| 59 |
|
| 60 |
To replicate images from Section 4.2 of the paper, you can use the following:
|
| 61 |
|