tsungtao commited on
Commit
cc61c95
·
1 Parent(s): 914fa69

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
22
  "tsungtao/controlnet-mlsd-202305011046", from_flax=True, dtype=jnp.bfloat16
23
  )
24
  pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
25
- "runwayml/stable-diffusion-v1-5/tree/main", controlnet=controlnet, revision="flax", dtype=jnp.bfloat16
26
  )
27
 
28
  def infer(prompts, negative_prompts, image):
 
22
  "tsungtao/controlnet-mlsd-202305011046", from_flax=True, dtype=jnp.bfloat16
23
  )
24
  pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
25
+ "runwayml/stable-diffusion-v1-5", controlnet=controlnet, revision="flax", dtype=jnp.bfloat16
26
  )
27
 
28
  def infer(prompts, negative_prompts, image):