Update README.md
Browse files
README.md
CHANGED
@@ -33,10 +33,10 @@ Weights for this model are available in Safetensors format.
|
|
33 |
## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
|
34 |
|
35 |
```py
|
36 |
-
from diffusers import
|
37 |
import torch
|
38 |
|
39 |
-
pipeline =
|
40 |
pipeline.load_lora_weights('Fer14/sdxl-coffe-machines', weight_name='pytorch_lora_weights.safetensors')
|
41 |
image = pipeline('A yellow espresso machine with portafilter handle and milk frother, box-shaped, with no screen and no buttons ').images[0]
|
42 |
```
|
|
|
33 |
## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
|
34 |
|
35 |
```py
|
36 |
+
from diffusers import DiffusionPipeline
|
37 |
import torch
|
38 |
|
39 |
+
pipeline = DiffusionPipeline.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', torch_dtype=torch.float16).to('cuda')
|
40 |
pipeline.load_lora_weights('Fer14/sdxl-coffe-machines', weight_name='pytorch_lora_weights.safetensors')
|
41 |
image = pipeline('A yellow espresso machine with portafilter handle and milk frother, box-shaped, with no screen and no buttons ').images[0]
|
42 |
```
|