Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
)
|