Cognomen commited on
Commit
96b24ec
Β·
1 Parent(s): dbd277d

forgot return

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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):