nousr commited on
Commit
20391f2
1 Parent(s): c36d220

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -22,7 +22,7 @@ import torch
22
  from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
23
 
24
  scheduler = EulerDiscreteScheduler.from_pretrained("nousr/robo-diffusion-2-base", subfolder="scheduler")
25
- pipe = StableDiffusionPipeline.from_pretrained("nousr/robo-diffusion-2-base", scheduler=scheduler, torch_dtype=torch.float16, revision="fp16")
26
  pipe = pipe.to("cuda")
27
 
28
  prompt = "A realistic photograph of a 3d nousr robot in a modern city. A glossy white and orange nousr robot."
22
  from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
23
 
24
  scheduler = EulerDiscreteScheduler.from_pretrained("nousr/robo-diffusion-2-base", subfolder="scheduler")
25
+ pipe = StableDiffusionPipeline.from_pretrained("nousr/robo-diffusion-2-base", scheduler=scheduler, torch_dtype=torch.float16)
26
  pipe = pipe.to("cuda")
27
 
28
  prompt = "A realistic photograph of a 3d nousr robot in a modern city. A glossy white and orange nousr robot."