feelatkeen commited on
Commit
88f326c
1 Parent(s): 3201d59

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -47,10 +47,10 @@ Weights for this model are available in Safetensors format.
47
 
48
  ```py
49
 
50
- from diffusers import DiffusionPipeline
51
  import torch
52
 
53
- pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
54
  pipeline.load_lora_weights("artificialguybr/ps1redmond-ps1-game-graphics-lora-for-sdxl")
55
  pipeline.enable_model_cpu_offload()
56
  init_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")
 
47
 
48
  ```py
49
 
50
+ from diffusers import AutoPipelineForImage2Image
51
  import torch
52
 
53
+ pipeline = AutoPipelineForImage2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
54
  pipeline.load_lora_weights("artificialguybr/ps1redmond-ps1-game-graphics-lora-for-sdxl")
55
  pipeline.enable_model_cpu_offload()
56
  init_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")