TheRafal commited on
Commit
dedc0d9
1 Parent(s): 00dffb8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -7
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("test.png")
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
  ----