Spaces:

jax-diffusers-event
/

canny_coyo1m

Paused

merve HF Staff commited on Apr 13, 2023

Commit

1c5616f

1 Parent(s): 7502a3f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,10 +35,10 @@ def canny_filter(image):
 # load control net and stable diffusion v1-5
 controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
-    "jax-diffusers-event/canny-coyo1m", dtype=jnp.bfloat16
 )
 pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5", controlnet=controlnet, from_pt=True, dtype=jnp.bfloat16
 )
 def infer(prompts, negative_prompts, image):

 # load control net and stable diffusion v1-5
 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):