Spaces:

jax-diffusers-event
/

canny_coyo1m

App Files Files Community

merve HF staff commited on Apr 12, 2023

Commit

a7c611e

•

1 Parent(s): c655d4d

Update app.py

Files changed (1) hide show

app.py +0 -1

app.py CHANGED Viewed

@@ -38,7 +38,6 @@ def canny_filter(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
 )

 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
 )