ehsanakh commited on
Commit
e1aa75f
1 Parent(s): 5e25c83

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -56,16 +56,16 @@ from diffusers import DiffusionPipeline
56
  import torch
57
 
58
  pipe = DiffusionPipeline.from_pretrained(
59
- "playgroundai/playground-v2-1024px-aesthetic",
60
  torch_dtype=torch.float16,
61
  use_safetensors=True,
62
  add_watermarker=False,
63
- variant="fp16"
64
  )
65
  pipe.to("cuda")
66
 
67
  prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
68
- image = pipe(prompt=prompt).images[0]
69
  ```
70
 
71
  ### User Study
 
56
  import torch
57
 
58
  pipe = DiffusionPipeline.from_pretrained(
59
+ "playgroundai/playground-v2-512px-base",
60
  torch_dtype=torch.float16,
61
  use_safetensors=True,
62
  add_watermarker=False,
63
+ variant="fp16",
64
  )
65
  pipe.to("cuda")
66
 
67
  prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
68
+ image = pipe(prompt=prompt, width=512, height=512).images[0]
69
  ```
70
 
71
  ### User Study