patrickvonplaten ameerazam08 commited on
Commit
c9822f0
1 Parent(s): c35c0a0

Update README.md (#5)

Browse files

- Update README.md (ace5058aafb5f81d4d9f60b8db0d449c626d7aea)


Co-authored-by: Ameer Azam <ameerazam08@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -63,7 +63,7 @@ You can use the model then as follows
63
  from diffusers import DiffusionPipeline
64
  import torch
65
 
66
- pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1-0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
67
  pipe.to("cuda")
68
 
69
  # if using torch < 2.0
 
63
  from diffusers import DiffusionPipeline
64
  import torch
65
 
66
+ pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
67
  pipe.to("cuda")
68
 
69
  # if using torch < 2.0