nateraw commited on
Commit
d1a77c4
·
1 Parent(s): 2ca681d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -23,7 +23,10 @@ To use this model, you can run the following...
23
  import torch
24
  from diffusers import StableDiffusionPipeline
25
 
26
- pipe = StableDiffusionPipeline.from_pretrained("nateraw/zach-fox", torch_dtype=torch.float16).to("cuda")
 
 
 
27
 
28
  prompt = "a photo of sks zach fox, oil on canvas"
29
  image = pipe(prompt).images[0]
 
23
  import torch
24
  from diffusers import StableDiffusionPipeline
25
 
26
+ pipe = StableDiffusionPipeline.from_pretrained(
27
+ "nateraw/zach-fox",
28
+ torch_dtype=torch.float16
29
+ ).to("cuda")
30
 
31
  prompt = "a photo of sks zach fox, oil on canvas"
32
  image = pipe(prompt).images[0]