Spaces:

jax-diffusers-event
/

canny_coyo1m

App Files Files Community

merve HF staff commited on Apr 12, 2023

Commit

c655d4d

•

1 Parent(s): b820c8c

Update app.py

Files changed (1) hide show

app.py +2 -0

app.py CHANGED Viewed

@@ -38,9 +38,11 @@ def canny_filter(image):
 controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
     "jax-diffusers-event/canny-coyo1m", dtype=jnp.float32
 )
 pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5", controlnet=controlnet, from_pt=True, dtype=jnp.float32
 )
 def infer(prompts, negative_prompts, image):

 controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
     "jax-diffusers-event/canny-coyo1m", dtype=jnp.float32
 )
+controlnet = controlnet.to("cuda")
 pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5", controlnet=controlnet, from_pt=True, dtype=jnp.float32
 )
+pipe = pipe.to("cuda")
 def infer(prompts, negative_prompts, image):