Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ pipeline, pipeline_params = FlaxStableDiffusionPipeline.from_pretrained(
|
|
12 |
)
|
13 |
|
14 |
|
15 |
-
def generate_image(prompt: str,negative_prompt:str , inference_steps: int = 25, prng_seed: int = 0, guidance_scale: float = 9):
|
16 |
rng = jax.random.PRNGKey(int(prng_seed))
|
17 |
rng = jax.random.split(rng, jax.device_count())
|
18 |
p_params = replicate(pipeline_params)
|
|
|
12 |
)
|
13 |
|
14 |
|
15 |
+
def generate_image(prompt: str,negative_prompt:str = "" , inference_steps: int = 25, prng_seed: int = 0, guidance_scale: float = 9):
|
16 |
rng = jax.random.PRNGKey(int(prng_seed))
|
17 |
rng = jax.random.split(rng, jax.device_count())
|
18 |
p_params = replicate(pipeline_params)
|