tsungtao commited on
Commit
d9076eb
·
1 Parent(s): 0e9f2e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -21,6 +21,9 @@ def canny_filter(image):
21
  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", controlnet=controlnet, revision="flax", dtype=jnp.bfloat16
26
  )
 
21
  controlnet, controlnet_params = FlaxControlNetModel.from_pretrained(
22
  "tsungtao/controlnet-mlsd-202305011046", from_flax=True, dtype=jnp.bfloat16
23
  )
24
+
25
+ controlnet.save_pretrained("tsungtao/controlnet-mlsd-202305011046")
26
+
27
  pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
28
  "runwayml/stable-diffusion-v1-5", controlnet=controlnet, revision="flax", dtype=jnp.bfloat16
29
  )