merve HF staff commited on
Commit
6ff2199
1 Parent(s): 31c6fcb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
38
  "jax-diffusers-event/canny-coyo1m", dtype=jnp.float32
39
  )
40
  pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
41
- "runwayml/stable-diffusion-v1-5", controlnet=controlnet, dtype=jnp.float32
42
  )
43
 
44
  def infer(prompts, negative_prompts, image):
 
38
  "jax-diffusers-event/canny-coyo1m", dtype=jnp.float32
39
  )
40
  pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
41
+ "runwayml/stable-diffusion-v1-5", from_pt=True, controlnet=controlnet, dtype=jnp.float32
42
  )
43
 
44
  def infer(prompts, negative_prompts, image):