Spaces:

Cognomen
/

CatCon-Controlnet-WD-1-5-b2

Runtime error

Cognomen commited on May 3, 2023

Commit

1e12c79

1 Parent(s): 96b24ec

fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -62,7 +62,7 @@ def infer(prompt, negative_prompt, image):
     n_prompt_in = shard(n_prompt_in)
     rng = get_random(0)
-    rng.random.split(rng, jax.device_count())
     p_params = replicate(params)

     n_prompt_in = shard(n_prompt_in)
     rng = get_random(0)
+    rng = jax.random.split(rng, jax.device_count())
     p_params = replicate(params)