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):
|