Update README.md
Browse files
README.md
CHANGED
@@ -66,28 +66,21 @@ Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.
|
|
66 |
|
67 |
## Trigger words
|
68 |
|
69 |
-
You should use an icon of trpfrog to trigger the image generation.
|
70 |
|
71 |
-
##
|
72 |
-
|
73 |
-
Weights for this model are available in Safetensors format.
|
74 |
-
|
75 |
-
[Download](Prgckwb/trpfrog-sdxl-lora-3/tree/main) them in the Files & versions tab.
|
76 |
-
|
77 |
-
|
78 |
-
|
79 |
-
## Intended uses & limitations
|
80 |
-
|
81 |
-
#### How to use
|
82 |
|
83 |
```python
|
84 |
-
|
85 |
-
|
86 |
|
87 |
-
|
|
|
88 |
|
89 |
-
|
|
|
|
|
90 |
|
91 |
-
|
92 |
-
|
93 |
-
|
|
|
66 |
|
67 |
## Trigger words
|
68 |
|
69 |
+
You should use an icon of "trpfrog" to trigger the image generation.
|
70 |
|
71 |
+
## How to use
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
72 |
|
73 |
```python
|
74 |
+
from diffusers import DiffusionPipeline
|
75 |
+
import torch
|
76 |
|
77 |
+
base_model_id = 'stabilityai/stable-diffusion-xl-base-1.0'
|
78 |
+
lora_model_id = 'Prgckwb/trpfrog-diffusion-sdxl-lora'
|
79 |
|
80 |
+
pipe = DiffusionPipeline.from_pretrained(base_model_id, torch_dtype=torch.float16)
|
81 |
+
pipe = pipe.to("cuda")
|
82 |
+
pipe.load_lora_weights(lora_model_id)
|
83 |
|
84 |
+
image = pipe("trpfrog", num_inference_steps=25).images[0]
|
85 |
+
image.save('trpfrog.png')
|
86 |
+
```
|