Update README.md
Browse files
README.md
CHANGED
@@ -21,8 +21,9 @@ This model was created as part of the DreamBooth Hackathon 🔥. Visit the [orga
|
|
21 |
|
22 |
## Description
|
23 |
|
|
|
|
|
24 |
|
25 |
-
This is a Stable Diffusion model fine-tuned on `` images for the wildcard theme.
|
26 |
|
27 |
|
28 |
## Usage
|
@@ -30,7 +31,21 @@ This is a Stable Diffusion model fine-tuned on `` images for the wildcard theme.
|
|
30 |
```python
|
31 |
from diffusers import StableDiffusionPipeline
|
32 |
|
33 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
34 |
image = pipeline().images[0]
|
35 |
-
|
36 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
|
22 |
## Description
|
23 |
|
24 |
+
This is a fine-tuned model of Stable Diffusion with the powerful help of Dreambooth.
|
25 |
+
The model is made specifically for TrpFrog ([@Trpfrog](https://twitter.com/trpfrog?s=21&t=GcDGy74adYhOBYxX9HloOg)) and faithfully reproduces its appearance.
|
26 |
|
|
|
27 |
|
28 |
|
29 |
## Usage
|
|
|
31 |
```python
|
32 |
from diffusers import StableDiffusionPipeline
|
33 |
|
34 |
+
keyword = "trpfrog"
|
35 |
+
prompt = f"a photo of {keyword}"
|
36 |
+
|
37 |
+
model_id = "Prgckwb/trpfrog-diffusion"
|
38 |
+
|
39 |
+
pipe = StableDiffusionPipeline.from_pretrained(
|
40 |
+
model_id, torch_dtype=torch.float16
|
41 |
+
).to(device)
|
42 |
+
|
43 |
image = pipeline().images[0]
|
44 |
+
```
|
45 |
+
|
46 |
+
## Generated Images
|
47 |
+
**a photo of trpfrog eating kiwi**
|
48 |
+
![](output.png)
|
49 |
+
|
50 |
+
**a photo of trpfrog growing shiitake mushrooms**
|
51 |
+
![](output2.png)
|