Spaces:
Runtime error
Runtime error
forgot return
Browse files
app.py
CHANGED
|
@@ -40,7 +40,7 @@ pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
|
| 40 |
#pipe.enable_xformers_memory_efficient_attention()
|
| 41 |
|
| 42 |
def get_random(seed):
|
| 43 |
-
jax.random.PRNGKey(seed)
|
| 44 |
|
| 45 |
# inference function takes prompt, negative prompt and image
|
| 46 |
def infer(prompt, negative_prompt, image):
|
|
|
|
| 40 |
#pipe.enable_xformers_memory_efficient_attention()
|
| 41 |
|
| 42 |
def get_random(seed):
|
| 43 |
+
return jax.random.PRNGKey(seed)
|
| 44 |
|
| 45 |
# inference function takes prompt, negative prompt and image
|
| 46 |
def infer(prompt, negative_prompt, image):
|