Fer14 commited on
Commit
714af3b
1 Parent(s): 61cd92a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 AutoPipelineForText2Image
37
  import torch
38
 
39
- pipeline = AutoPipelineForText2Image.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
  ```
 
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
  ```