Spaces:

G-AshwinKumar
/

stable-diffusion-nano

Paused

bguisard commited on Apr 27, 2023

Commit

971eeee

•

1 Parent(s): dac85aa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ pipeline, pipeline_params = FlaxStableDiffusionPipeline.from_pretrained(
 def generate_image(prompt: str, inference_steps: int, prng_seed: int):
-    rng = jax.random.PRNGKey(int(prng_seed))
     prompt_ids = pipeline.prepare_inputs(prompt)
     images = pipeline(

 def generate_image(prompt: str, inference_steps: int, prng_seed: int):
+    rng = jax.random.PRNGKey(prng_seed)
     prompt_ids = pipeline.prepare_inputs(prompt)
     images = pipeline(