Update README.md
Browse files
README.md
CHANGED
@@ -63,16 +63,13 @@ import torch
|
|
63 |
from torch import autocast
|
64 |
from diffusers import StableDiffusionPipeline
|
65 |
|
66 |
-
pipe = StableDiffusionPipeline.from_pretrained(
|
67 |
-
'TheRafal/everything-v1',
|
68 |
-
torch_dtype=torch.float32
|
69 |
-
).to('cuda')
|
70 |
|
71 |
-
prompt = "1girl, monika \(doki doki literature club\)"
|
72 |
with autocast("cuda"):
|
73 |
-
image = pipe(prompt, guidance_scale=7)["images"][0]
|
74 |
|
75 |
-
image.save("
|
76 |
```
|
77 |
|
78 |
----
|
|
|
63 |
from torch import autocast
|
64 |
from diffusers import StableDiffusionPipeline
|
65 |
|
66 |
+
pipe = StableDiffusionPipeline.from_pretrained('TheRafal/everything-v1', torch_dtype=torch.float32).to('cuda')
|
|
|
|
|
|
|
67 |
|
68 |
+
prompt = "1girl, monika \(doki doki literature club\), indoors, classroom"
|
69 |
with autocast("cuda"):
|
70 |
+
image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5)["images"][0]
|
71 |
|
72 |
+
image.save(prompt.replace(" ", "_") + ".png")
|
73 |
```
|
74 |
|
75 |
----
|