Spaces:

jax-diffusers-event
/

canny_coyo1m

Paused

merve HF Staff commited on Apr 13, 2023

Commit

8e80da4

1 Parent(s): eb568aa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,9 +46,9 @@ def infer(prompts, negative_prompts, image):
     params["controlnet"] = controlnet_params
-    num_samples = jax.device_count("GPU")
     rng = create_key(0)
-    rng = jax.random.split(rng, jax.device_count("GPU"))
     im = canny_filter(image)
     canny_image = Image.fromarray(im)

     params["controlnet"] = controlnet_params
+    num_samples = jax.device_count("gpu")
     rng = create_key(0)
+    rng = jax.random.split(rng, jax.device_count("gpu"))
     im = canny_filter(image)
     canny_image = Image.fromarray(im)