Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -140,7 +140,7 @@ pipeline, params = FlaxStableDiffusionPipeline.from_pretrained(
140
  prompt = "a photo of an astronaut riding a horse on mars"
141
 
142
  prng_seed = jax.random.PRNGKey(0)
143
- num_inference_steps = 50
144
 
145
  num_samples = jax.device_count()
146
  prompt = num_samples * [prompt]
@@ -173,7 +173,7 @@ pipeline, params = FlaxStableDiffusionPipeline.from_pretrained(
173
  prompt = "a photo of an astronaut riding a horse on mars"
174
 
175
  prng_seed = jax.random.PRNGKey(0)
176
- num_inference_steps = 50
177
 
178
  num_samples = jax.device_count()
179
  prompt = num_samples * [prompt]
 
140
  prompt = "a photo of an astronaut riding a horse on mars"
141
 
142
  prng_seed = jax.random.PRNGKey(0)
143
+ num_inference_steps = 150
144
 
145
  num_samples = jax.device_count()
146
  prompt = num_samples * [prompt]
 
173
  prompt = "a photo of an astronaut riding a horse on mars"
174
 
175
  prng_seed = jax.random.PRNGKey(0)
176
+ num_inference_steps = 150
177
 
178
  num_samples = jax.device_count()
179
  prompt = num_samples * [prompt]