nousr
/

robo-diffusion-2-base

StableDiffusionPipeline

stable-diffusion

Model card Files Files and versions

nousr commited on Nov 29, 2022

Commit

20391f2

·

1 Parent(s): c36d220

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -22,7 +22,7 @@ import torch
 from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
 scheduler = EulerDiscreteScheduler.from_pretrained("nousr/robo-diffusion-2-base", subfolder="scheduler")
-pipe = StableDiffusionPipeline.from_pretrained("nousr/robo-diffusion-2-base", scheduler=scheduler, torch_dtype=torch.float16, revision="fp16")
 pipe = pipe.to("cuda")
 prompt = "A realistic photograph of a 3d nousr robot in a modern city. A glossy white and orange nousr robot."

 from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
 scheduler = EulerDiscreteScheduler.from_pretrained("nousr/robo-diffusion-2-base", subfolder="scheduler")
+pipe = StableDiffusionPipeline.from_pretrained("nousr/robo-diffusion-2-base", scheduler=scheduler, torch_dtype=torch.float16)
 pipe = pipe.to("cuda")
 prompt = "A realistic photograph of a 3d nousr robot in a modern city. A glossy white and orange nousr robot."