Update README.md
Browse files
README.md
CHANGED
@@ -21,7 +21,7 @@ Keep the words `nousr robot` towards the beginning of your prompt to invoke the
|
|
21 |
import torch
|
22 |
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
|
23 |
|
24 |
-
scheduler = EulerDiscreteScheduler.from_pretrained(
|
25 |
pipe = StableDiffusionPipeline.from_pretrained("nousr/robo-diffusion-2-base", scheduler=scheduler, torch_dtype=torch.float16, revision="fp16")
|
26 |
pipe = pipe.to("cuda")
|
27 |
|
|
|
21 |
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 |
|