Update README.md
Browse files
README.md
CHANGED
|
@@ -15,8 +15,12 @@ Here we give an example model fine-tuned using 5 images of a cat downloaded from
|
|
| 15 |
|
| 16 |
|
| 17 |
## Example code of inference
|
| 18 |
-
```
|
|
|
|
|
|
|
| 19 |
|
|
|
|
|
|
|
| 20 |
from diffusers import StableDiffusionPipeline
|
| 21 |
from src import diffuser_training
|
| 22 |
|
|
@@ -31,5 +35,5 @@ prompt = "<new1> cat swimming in a pool"
|
|
| 31 |
images = pipe(prompt, num_inference_steps=200, guidance_scale=6., eta=1.).images
|
| 32 |
```
|
| 33 |
|
| 34 |
-
<img src='cat.png' align="center">
|
| 35 |
|
|
|
|
| 15 |
|
| 16 |
|
| 17 |
## Example code of inference
|
| 18 |
+
```
|
| 19 |
+
git clone https://github.com/adobe-research/custom-diffusion
|
| 20 |
+
cd custom-diffusion
|
| 21 |
|
| 22 |
+
```
|
| 23 |
+
```python
|
| 24 |
from diffusers import StableDiffusionPipeline
|
| 25 |
from src import diffuser_training
|
| 26 |
|
|
|
|
| 35 |
images = pipe(prompt, num_inference_steps=200, guidance_scale=6., eta=1.).images
|
| 36 |
```
|
| 37 |
|
| 38 |
+
<img src='https://huggingface.co/nupurkmr9/custom_diffusion_cat/blob/main/cat.png' align="center">
|
| 39 |
|