Update README.md
Browse files
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
|
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."
|