Spaces:

jax-diffusers-event
/

canny_coyo1m

Paused

merve HF Staff commited on Apr 13, 2023

Commit

6ff2199

1 Parent(s): 31c6fcb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ 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, dtype=jnp.float32
 )
 def infer(prompts, negative_prompts, image):

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