junnyu commited on
Commit
9c1ba7f
1 Parent(s): 6d9d6ba

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -71,9 +71,9 @@ import paddle
71
  from paddle.amp import auto_cast as autocast
72
  from diffusers_paddle import StableDiffusionPipeline
73
 
74
- model_id = "CompVis/stable-diffusion-v1-4"
75
 
76
- pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True)
77
 
78
  prompt = "a photo of an astronaut riding a horse on mars"
79
  with autocast(True):
@@ -89,7 +89,7 @@ If you are limited by GPU memory and have less than 10GB of GPU RAM available, p
89
  ```py
90
  import paddle
91
 
92
- pipe = StableDiffusionPipeline.from_pretrained(model_id, paddle_dtype=paddle.float16, use_auth_token=True)
93
 
94
  prompt = "a photo of an astronaut riding a horse on mars"
95
  with autocast(True):
@@ -106,7 +106,7 @@ from paddle.amp import auto_cast as autocast
106
  model_id = "junnyu/stable-diffusion-v1-4-paddle"
107
  # Use the K-LMS scheduler here instead
108
  scheduler = LMSDiscreteScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000)
109
- pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, use_auth_token=True)
110
 
111
  prompt = "a photo of an astronaut riding a horse on mars"
112
  with autocast(True):
 
71
  from paddle.amp import auto_cast as autocast
72
  from diffusers_paddle import StableDiffusionPipeline
73
 
74
+ model_id = "junnyu/stable-diffusion-v1-4-paddle"
75
 
76
+ pipe = StableDiffusionPipeline.from_pretrained(model_id)
77
 
78
  prompt = "a photo of an astronaut riding a horse on mars"
79
  with autocast(True):
 
89
  ```py
90
  import paddle
91
 
92
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, paddle_dtype=paddle.float16)
93
 
94
  prompt = "a photo of an astronaut riding a horse on mars"
95
  with autocast(True):
 
106
  model_id = "junnyu/stable-diffusion-v1-4-paddle"
107
  # Use the K-LMS scheduler here instead
108
  scheduler = LMSDiscreteScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000)
109
+ pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler)
110
 
111
  prompt = "a photo of an astronaut riding a horse on mars"
112
  with autocast(True):